Rename app.py to app_test.py

app.py → app_test.py

@@ -58,12 +58,15 @@ with gr.Blocks() as demo:
            
         with gr.Row():
             output = [gr.Image(label = m) for m in default_models]
+            current_models = default_models
+            
             model_choice.change(update_imgbox, model_choice, output)
+            model_choice.change(extend_choices, model_choice, current_models, preprocess = False, postprocess = False)
 
-            #for o in output:
-            #    gen_button.click(lambda t: gen_fn(o.label, t), txt_input, o)
+            for m, o in zip(current_models, output):
+                gen_button.click(lambda txt: gen_fn(m, txt), txt_input, o)
                     
-            gen_button.click(gen_fn2, [model_choice, txt_input], output, batch = True, max_batch_size = num_models)
+            #gen_button.click(gen_fn2, [model_choice, txt_input], output, batch = True, max_batch_size = num_models)
            
             '''
             output = gr.Gallery()