Update app.py

app.py

@@ -58,6 +58,10 @@ def extend_choices(choices):
     return choices + (nb_models - len(choices)) * ['NA']
     """return choices + (num_models - len(choices)) * ['NA']"""
 
+def extend_choices_b(choices):
+    choices_plus = extend_choices(choices)
+    return [gr.Textbox(m, visible=False) for m in choices_plus]
+
 def update_imgbox(choices):
     choices_plus = extend_choices(choices)
     return [gr.Image(None, label=m, visible=(m != 'NA')) for m in choices_plus]
@@ -121,7 +125,7 @@ def make_me():
             """output = [gr.Image(label=m, min_width=170, height=170) for m in choices]
             current_models = [gr.Textbox(m, visible=False) for m in choices]"""
             output = update_imgbox([choices[0][0]])
-            current_models = extend_choices([choices[0][0]])
+            current_models = extend_choices_b([choices[0][0]])
             
             for m, o in zip(current_models, output):
                 gen_event = gen_button.click(gen_fn, [m, txt_input], o)