Update app.py

app.py

@@ -10,7 +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()  # 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