Update app.py

app.py

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