Update app.py

app.py

@@ -68,8 +68,7 @@ with gr.Blocks() as demo:
         
         with gr.Column():
             num_images = gr.Slider(1, max_images, value = 1, step = 1, label = 'Number of images')        
-            num_models = num_images
-            output2 = [gr.Image(label = '') for _ in range(num_models)]
+            output2 = [gr.Image(label = '') for _ in range(6)]
         for i, o in enumerate(output2):
             num_models = num_images
             img_i = gr.Number(i, visible = False)