Update app.py

app.py

@@ -320,6 +320,13 @@ def video_generation_handler(prompt, seed=42, fps=15):
         yield None, None, error_status_html
 
 # --- Gradio UI Layout ---
+frame_display = gr.Image(
+    label="Generated Frames",
+    height=480,
+    width=832,
+    show_label=True,
+    container=True
+)
 with gr.Blocks(title="Self-Forcing Frame Streaming Demo") as demo:
     gr.Markdown("# 🚀 Self-Forcing Video Generation with Frame Streaming")
     gr.Markdown("Real-time video generation with frame-by-frame display. [[Model]](https://huggingface.co/gdhe17/Self-Forcing), [[Project page]](https://self-forcing.github.io), [[Paper]](https://huggingface.co/papers/2506.08009)")
@@ -363,13 +370,7 @@ with gr.Blocks(title="Self-Forcing Frame Streaming Demo") as demo:
             gr.Markdown("### 📺 Live Frame Stream")
             gr.Markdown("*Click 'Start Generation' to begin frame streaming*")
             
-            frame_display = gr.Image(
-                label="Generated Frames",
-                height=480,
-                width=832,
-                show_label=True,
-                container=True
-            )
+            frame_display.render()
             
             final_video = gr.Video(
                 label="Final Rendered Video",