Update app.py

app.py

@@ -53,17 +53,13 @@ def generate(num_images, interpolate):
 ####################################################################
 # Graphical User Interface
 
-#demo = gr.Blocks()
+demo = gr.Blocks()
 
 def infer(num_images, interpolate):
     img = generate(round(num_images), interpolate)
     imgs = list(img)
     return imgs
 
-demo = gr.Blocks()
-
-output = gr.Gallery()
-    
 with demo:
     gr.Markdown(
     """
@@ -74,7 +70,9 @@ with demo:
     interpolate = gr.inputs.Checkbox(default=False, label="Interpolate")
     submit = gr.Button("Generate")
     
-    out = output
+    
+    out = gr.Gallery()
+
     submit.click(fn=infer, 
                inputs=[images_num, interpolate], 
                outputs=out)