Update app.py

app.py

@@ -51,8 +51,7 @@ with gr.Blocks() as demo:
         gen_button = gr.Button('Generate')
            
         with gr.Row():
-            output = [gr.Image(label = m) for m in default_models]
-            current_models = [gr.Textbox(m) for m in default_models]
+            output, current_models = unzip([gr.Image(label = m), gr.Textbox(m, visible = False) for m in default_models]]
             
             model_choice.change(update_imgbox, model_choice, output)
             model_choice.change(extend_choices, model_choice, current_models)