guidance

app.py

@@ -46,11 +46,11 @@ W, H = 512, 320  # the code calls pipeline with width=512, height=320
 # --------- action types ----------
 # Display labels mapped to integer IDs
 ACTION_TYPES = {
-    "normal (0)": 0,
-    "ego (1)": 1,
-    "ego/veh (2)": 2,
-    "veh (3)": 3,
-    "veh/veh (4)": 4,
+    "No Crash": 0,
+    "Ego-Only crash": 1,
+    "Ego-and-Vehicle crash": 2,
+    "Vehicle-Only crash": 3,
+    "Vehicle-and-Vehicle crash": 4,
 }
 
 # --------- helpers ----------
@@ -137,7 +137,7 @@ def make_sample(
     }
     return sample
 
-g_use_factor_guidance = True
+g_use_factor_guidance = False
 
 pipe = load_ctrlv_pipelines(
     "AnthonyGosselin/Ctrl-Crash",
@@ -345,41 +345,47 @@ with gr.Blocks(title="Ctrl-Crash Demo") as demo:
     gr.Markdown("# Ctrl-Crash: Image → Conditioned Video")
     with gr.Row():
         with gr.Column(scale=1):
-            video_in = gr.Video(label="Init Video (will be letterboxed to 512×320)")
+            video_in = gr.Video(label="Dashcam Footage")
 
             action_dropdown = gr.Dropdown(
                 label="Action Type",
                 choices=list(ACTION_TYPES.keys()),
-                value="ego/veh (2)",  # sensible default matching your prior checkbox default
+                value="Vehicle-and-Vehicle crash",  # sensible default matching your prior checkbox default
             )
             bbx_frames = gr.Slider(1, 25, value=3, step=1, label="Frames")
             run_btn = gr.Button("Generate Video", variant="primary")
         with gr.Column(scale=1):
-            video_out = gr.Video(label="Output video (MP4)")
+            video_out = gr.Video(label="Output video")
 
             cached_examples = gr.Examples(                    
                 examples=[
                     [
                         "examples/gt_video.mp4",
-                        "normal (0)",
+                        "No Crash",
                         3,
                     ],
 
                     [
                         "examples/gt_video.mp4",
-                        "ego (1)",
+                        "Ego-Only crash",
                         3,
                     ],
 
                     [
                         "examples/gt_video.mp4",
-                        "veh (3)",
+                        "Ego-and-Vehicle crash",
                         3,
                     ],
 
                     [
                         "examples/gt_video.mp4",
-                        "veh/veh (4)",
+                        "Vehicle-Only crash",
+                        3,
+                    ],
+
+                    [
+                        "examples/gt_video.mp4",
+                        "Vehicle-and-Vehicle crash",
                         3,
                     ],