Update app.py

app.py

@@ -11,7 +11,7 @@ lock = RLock()
 HF_TOKEN = os.environ.get("HF_TOKEN") if os.environ.get("HF_TOKEN") else None # If private or gated models aren't used, ENV setting is unnecessary.
 # --- Step 2: Authenticate and fetch your models
 
-default_models = models[:num_models]
+
 api = HfApi()
 user_info = whoami(token=HF_TOKEN)
 username = user_info["name"]
@@ -26,23 +26,9 @@ def random_choices():
     import random
     random.seed()
     return random.choices(models, k=num_models)
-mymodels = list(api.list_models(author=username, token=HF_TOKEN))
-model_ids = [m.modelId for m in mymodels]
-if not model_ids:
-    raise ValueError(f"No models found for user '{username}'")
-# --- Step 3: Build Gradio UI
-def handle_model_selection(selected_models):
-    if not selected_models:
-        return "No models selected."
-    return "✅ Selected models:\n" + "\n".join(selected_models)
-
-
-
 
 
-
-
-load_fn(default_models,HF_TOKEN)
+load_fn(models,HF_TOKEN)
 
 
 '''
@@ -52,11 +38,6 @@ load_fn(default_models,HF_TOKEN)
 
 
 with gr.Blocks(fill_width=True) as demo:
-    with gr.Row():
-        gr.Markdown(f"# ({username}) you are logged in")
-        model_selector = gr.CheckboxGroup(choices=model_ids,value=model_ids, label="your models",        interactive=True,    )
-        output_box = gr.Textbox(lines=10, label="Selected Models")
-        model_selector.change(fn=handle_model_selection, inputs=model_selector, outputs=output_box)   
     with gr.Tab(str(num_models) + ' Models'):
         with gr.Column(scale=2):
             with gr.Group():
@@ -92,7 +73,7 @@ with gr.Blocks(fill_width=True) as demo:
 
         with gr.Column(scale=4):
             with gr.Accordion('Model selection'):
-                model_choice = gr.CheckboxGroup(default_models, label = f'Choose up to {int(num_models)} different models from the {len(default_models)} available!', value=default_models, interactive=True)
+                model_choice = gr.CheckboxGroup(models, label = f'Choose up to {int(num_models)} different models from the {len(models)} available!', value=models, interactive=True)
                 model_choice.change(update_imgbox, model_choice, output)
                 model_choice.change(extend_choices, model_choice, current_models)
                 random_button.click(random_choices, None, model_choice)