Update app.py

app.py

@@ -6,15 +6,15 @@ from utils import generate_gradio_images
 iface = gr.Interface(
     fn=generate_gradio_images,  # Set to None since we won't use a function
     inputs=[
-        gr.Textbox("Enter Prompt"),
+        gr.Textbox(),
         gr.Slider("Time Stamps", value=1, step=1, maximum=50),
         gr.Checkbox(label = "Use custom loss",value=False)
     ],
-    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
+    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
 )
 
 # Launch the Gradio interface