Update app.py

app.py

@@ -85,6 +85,11 @@ def make_me():
         with gr.Row():
             output = [gr.Image(label=m, min_width=170, height=170) for m in default_models]
             current_models = [gr.Textbox(m, visible=False) for m in default_models]
+            
+            choices=models_test[0]
+            output=update_imgbox(choices)
+            current_models=extend_choices(choices)
+            
             for m, o in zip(current_models, output):
                 gen_event = gen_button.click(gen_fn, [m, txt_input], o)
                 stop_button.click(lambda s: gr.update(interactive=False), None, stop_button, cancels=[gen_event])