Update app.py

app.py

@@ -327,6 +327,17 @@ frame_display = gr.Image(
     show_label=True,
     container=True
 )
+final_video = gr.Video(
+    label="Final Rendered Video", 
+    visible=False, 
+    interactive=False,
+    height=400,
+    autoplay=True
+)
+status_html = gr.HTML(
+                value="<div style='text-align: center; padding: 20px; color: #666;'>Ready to start generation...</div>",
+                label="Generation Status"
+            )
 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)")
@@ -369,21 +380,12 @@ with gr.Blocks(title="Self-Forcing Frame Streaming Demo") as demo:
         with gr.Column(scale=3):
             gr.Markdown("### 📺 Live Frame Stream")
             gr.Markdown("*Click 'Start Generation' to begin frame streaming*")
+
+            final_video.render()
             
             frame_display.render()
-            
-            final_video = gr.Video(
-                label="Final Rendered Video", 
-                visible=False, 
-                interactive=False,
-                height=400,
-                autoplay=True
-            )
-            
-            status_html = gr.HTML(
-                value="<div style='text-align: center; padding: 20px; color: #666;'>Ready to start generation...</div>",
-                label="Generation Status"
-            )
+
+            status_html.render()
 
     # Connect the generator to the image display
     start_btn.click(