Update app.py

app.py

@@ -54,7 +54,9 @@ load_fn(models)"""
     
 
 def extend_choices(choices):
-    return choices + (num_models - len(choices)) * ['NA']
+    nb_models
+    return choices + (nb_models - len(choices)) * ['NA']
+    """return choices + (num_models - len(choices)) * ['NA']"""
 
 def update_imgbox(choices):
     choices_plus = extend_choices(choices)
@@ -116,8 +118,10 @@ def make_me():
             current_models = [gr.Textbox(m, visible=False) for m in default_models]"""
             """choices=[models_test[0][0]]"""
             choices=models_test[0]
-            output = [gr.Image(label=m, min_width=170, height=170) for m in choices]
-            current_models = [gr.Textbox(m, visible=False) for m in choices]
+            """output = [gr.Image(label=m, min_width=170, height=170) for m in choices]
+            current_models = [gr.Textbox(m, visible=False) for m in choices]"""
+            output = update_imgbox([choices[0][0]])
+            current_models = extend_choices([choices[0][0]])
             
             for m, o in zip(current_models, output):
                 gen_event = gen_button.click(gen_fn, [m, txt_input], o)
@@ -127,9 +131,7 @@ def make_me():
             model_choice.change(update_imgbox, (gen_button,stop_button,group_model_choice), output)
             model_choice.change(extend_choices, model_choice, current_models)
         """     
-        """with gr.Accordion('Group Model selection'):
-            group_model_choice = gr.CheckboxGroup(models_test, label=f' {len(models_test)} different group models', value=1, multiselect=False, max_choices=1, interactive=True, filterable=False)
-        """
+        
         with gr.Accordion("test", open=True):
             group_model_choice = gr.Dropdown(label="test Model", show_label=False, choices=list(models_test) , allow_custom_value=True)
             """group_model_choice = gr.Dropdown(label="test Model", show_label=False, choices=list([]) , allow_custom_value=True)"""