Update app.py

app.py

@@ -68,6 +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)]
         for i, o in enumerate(output2):
             num_models = num_images