Constrain generated rollout display size

app.py

@@ -539,6 +539,20 @@ def _latest_mp4_since(start_time: float) -> Path | None:
     return candidates[0] if candidates else None
 
 
+DEMO_CSS = """
+#generated-rollout {
+    width: min(100%, 420px) !important;
+    max-width: 420px !important;
+}
+
+#generated-rollout video {
+    width: 100% !important;
+    max-height: 240px !important;
+    object-fit: contain !important;
+}
+"""
+
+
 @spaces.GPU(duration=120)
 def run_inference(
     setting: str,
@@ -578,7 +592,7 @@ def run_inference(
     return str(mp4), _tail(log + f"\nGenerated video: {mp4.relative_to(ROOT)}")
 
 
-with gr.Blocks(title="PhysFormer") as demo:
+with gr.Blocks(title="PhysFormer", css=DEMO_CSS) as demo:
     gr.Markdown(
         """
         # PhysFormer Minimal ZeroGPU Demo
@@ -630,7 +644,7 @@ with gr.Blocks(title="PhysFormer") as demo:
         height=420,
     )
     run_button = gr.Button("Generate", variant="primary")
-    video = gr.Video(label="Generated rollout", height=320, width=560)
+    video = gr.Video(label="Generated rollout", height=240, width=420, elem_id="generated-rollout")
     log = gr.Textbox(label="Log", lines=18)
 
     setting.change(