Update app.py

app.py

@@ -60,10 +60,10 @@ with gr.Blocks() as demo:
             output = [gr.Image(label = m) for m in default_models]
             model_choice.change(update_imgbox, model_choice, output)
 
-            for o in output:
-                gen_button.click(lambda t: gen_fn(o.label, t), txt_input, o)
+            #for o in output:
+            #    gen_button.click(lambda t: gen_fn(o.label, t), txt_input, o)
                     
-            #gen_button.click(gen_fn2, [model_choice, txt_input], output)
+            gen_button.click(gen_fn2, [model_choice, txt_input], output, batch = True, max_batch_size = num_models)
            
             '''
             output = gr.Gallery()