Update app.py

app.py

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