Update app.py

app.py

@@ -22,7 +22,6 @@ load_fn(models)
 
 num_models = 4
 default_models = models[:num_models]
-#current_models = default_models
 
 
 
@@ -42,11 +41,6 @@ def gen_fn(model_str, prompt):
     return models_load[model_str](f'{prompt} {noise}')
 
 
-def gen_fn2(choices, prompt):
-    choices_plus = extend_choices(choices)
-    return [gen_fn(m, prompt) for m in choices_plus]
-
-
 
 with gr.Blocks() as demo:
     with gr.Tab('Multiple models'): 
@@ -66,12 +60,6 @@ with gr.Blocks() as demo:
             for m, o in zip(current_models, output):
                 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)
-           
-            '''
-            output = gr.Gallery()
-            gen_button.click(lambda models, txt: [gen_fn(m, txt) for m in models], [model_choice, txt_input], output)
-            '''
             
     with gr.Tab('Single model'):
         model_choice2 = gr.Dropdown(models, label = 'Choose model', value = models[0], filterable = False)