Update app.py

app.py

@@ -10,11 +10,11 @@ iface = gr.Interface(
         gr.Slider("Time Stamps", value=1, step=1, maximum=50),
         gr.Checkbox(label = "Use custom loss",value=False)
     ],
-    outputs=[gr.Image(type="pil", height=512, width = 512),
-             gr.Image(type="pil", height=512, width = 512), 
-             gr.Image(type="pil", height=512, width = 512), 
-             gr.Image(type="pil", height=512, width = 512),
-             gr.Image(type="pil", height=512, width = 512)]  # Gradio Image component for displaying images
+    outputs=[gr.Image(type="pil", height=128, width = 128),
+             gr.Image(type="pil", height=128, width = 128), 
+             gr.Image(type="pil", height=128, width = 128), 
+             gr.Image(type="pil", height=128, width = 128),
+             gr.Image(type="pil", height=128, width = 128)]  # Gradio Image component for displaying images
 )
 
 # Launch the Gradio interface