Update app.py

app.py

@@ -62,6 +62,8 @@ def infer(num_images, interpolate):
 
 demo = gr.Blocks()
 
+output = gr.Gallery()
+    
 with demo:
     gr.Markdown(
     """
@@ -72,8 +74,7 @@ with demo:
     interpolate = gr.inputs.Checkbox(default=False, label="Interpolate")
     submit = gr.Button("Generate")
     
-    out = gr.Gallery()
-
+    out = output
     submit.click(fn=infer, 
                inputs=[images_num, interpolate], 
                outputs=out)