Update app.py

app.py

@@ -20,7 +20,7 @@ def load_fn(models):
 load_fn(models)
 
 
-num_models = 4
+num_models = 6
 default_models = models[:num_models]
 
 
@@ -51,7 +51,7 @@ with gr.Blocks() as demo:
         gen_button = gr.Button('Generate')
            
         with gr.Row():
-            output, current_models = unzip([[gr.Image(label = m), gr.Textbox(m, visible = True)] for m in default_models])
+            output, current_models = list(zip(*[[gr.Image(label = m), gr.Textbox(m, visible = True)] for m in default_models]))
             
             model_choice.change(update_imgbox, model_choice, output)
             model_choice.change(extend_choices, model_choice, current_models)