Update app_test.py

app_test.py

@@ -25,13 +25,6 @@ default_models = models[:num_models]
 #current_models = default_models
 
 
-'''
-def make_update(choices):
-    global current_models
-    current_models = [c for c in choices]
-    current_models += (num_models - len(choices)) * ['NA']
-    return [gr.Image(None, label = m, visible = (m != 'NA')) for m in current_models]
-'''
 
 def extend_choices(choices):
     return choices + (num_models - len(choices)) * ['NA']
@@ -39,7 +32,7 @@ def extend_choices(choices):
 
 def update_imgbox(choices):
     choices_plus = extend_choices(choices)
-    return [gr.Image(None, label = m, visible = (m != 'NA')) for m in choices_plus]
+    return [gr.Image(None, label = m, visible = True) for m in choices_plus]
 
     
 def gen_fn(model_str, prompt):
@@ -67,7 +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)
 
-            gen_button.click(gen_fn2, [model_choice, txt_input], output)
+            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)
            
             '''
             output = gr.Gallery()