Update app.py

app.py

@@ -67,7 +67,7 @@ with gr.Blocks() as demo:
                )
         
         with gr.Row():
-            output2 = [gr.Image(label = '', min_width=480) for _ in range(max_images)]
+            output2 = [gr.Image(label = '', min_width=480) for _ in range(num_models)]
             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)