Update app.py

app.py

@@ -1,21 +1,52 @@
+# import gradio as gr
+# from PIL import Image
+# from utils import generate_gradio_images
+
+# # Create a Gradio interface
+# iface = gr.Interface(
+#     fn=generate_gradio_images,  # Set to None since we won't use a function
+#     inputs=[
+#         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=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
+# iface.launch()
+
 import gradio as gr
 from PIL import Image
 from utils import generate_gradio_images
 
+width, height = 512, 512
 # Create a Gradio interface
-iface = gr.Interface(
-    fn=generate_gradio_images,  # Set to None since we won't use a function
-    inputs=[
-        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=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
-)
+def gradio_app():
+    with gr.Blocks() as demo:
+        with gr.Column():
+            input_prompt = gr.Textbox(label="Enter Prompt")
+            input_slider = gr.Slider(label="Time Stamps", value=1, step=1, maximum=50)
+            input_checkbox = gr.Checkbox(label="Use custom loss", value=False)
+        
+        with gr.Row():
+            output_image1 = gr.Image(type="pil", height=height, width=width)
+            output_image2 = gr.Image(type="pil", height=height, width=width)
+            output_image3 = gr.Image(type="pil", height=height, width=width)
+            output_image4 = gr.Image(type="pil", height=height, width=width)
+            output_image5 = gr.Image(type="pil", height=height, width=width)
+        
+        inputs = [input_prompt, input_slider, input_checkbox]
+        outputs = [output_image1, output_image2, output_image3, output_image4, output_image5]
+        
+        gr.Interface(fn=generate_gradio_images, inputs=inputs, outputs=outputs)
+        
+    return demo
 
 # Launch the Gradio interface
-iface.launch()
+demo = gradio_app()
+demo.launch()