Update app.py

app.py

@@ -22,14 +22,29 @@ def generate_pil_images():
     pil_images = [Image.fromarray(np.uint8(image)) for image in generated_images]
     return pil_images
 
+# Path to the video of training images visualization
+training_video_path = "training_visualization.mp4"  # Replace with your video file path
+
 # Create a Gradio interface
-iface = gr.Interface(
-    fn=generate_pil_images,
-    inputs=[],
-    outputs=gr.Gallery(label="Generated Images",columns=4, height="fill"),
-    title="Generated Images",
-    description="This Gradio app generates 16 random images using a pre-trained generator model whenever the interface is refreshed."
-)
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("# Generated Images")
+            gr.Interface(
+                fn=generate_pil_images,
+                inputs=[],
+                outputs=gr.Gallery(label="Generated Images", columns=4, height="fill"),
+                live=True  # Live interface so it updates every time you refresh
+            )
+        with gr.Column():
+            gr.Markdown("# Training Visualization Video")
+            gr.Video(
+                value=training_video_path,
+                label="Training Visualization",
+                format="mp4",
+                autoplay=True,
+                loop=True
+            )
 
 # Launch the Gradio app
-iface.launch()
+demo.launch()