Update app.py

app.py

@@ -103,6 +103,11 @@ def handle(mode, media_upload, url,
 
     first_input = media[0]
 
+    # 🔧 Resolve dropdown label to model keys
+    resolved_det_model = DETECTION_MODEL_MAP.get(det_model, det_model)
+    resolved_seg_model = SEGMENTATION_MODEL_MAP.get(seg_model, seg_model)
+    resolved_depth_model = DEPTH_MODEL_MAP.get(depth_model, depth_model)
+
     # --- VIDEO PATH ---
     if isinstance(first_input, str) and first_input.lower().endswith((".mp4", ".mov", ".avi")):
         valid, err = validate_video(first_input)
@@ -117,12 +122,12 @@ def handle(mode, media_upload, url,
             _, msg, output_video_path = process_video(
                 video_path=first_input,
                 run_det=run_det,
-                det_model=det_model,
+                det_model=resolved_det_model,
                 det_confidence=det_confidence,
                 run_seg=run_seg,
-                seg_model=seg_model,
+                seg_model=resolved_seg_model,
                 run_depth=run_depth,
-                depth_model=depth_model,
+                depth_model=resolved_depth_model,
                 blend=blend
             )
             return (
@@ -154,12 +159,12 @@ def handle(mode, media_upload, url,
             result_img, msg, output_zip = process_image(
                 image=first_input,
                 run_det=run_det,
-                det_model=det_model,
+                det_model=resolved_det_model,
                 det_confidence=det_confidence,
                 run_seg=run_seg,
-                seg_model=seg_model,
+                seg_model=resolved_seg_model,
                 run_depth=run_depth,
-                depth_model=depth_model,
+                depth_model=resolved_depth_model,
                 blend=blend
             )
             return (
@@ -195,6 +200,7 @@ def handle(mode, media_upload, url,
     )
 
 
+
 def show_preview_from_upload(files):
     if not files:
         return gr.update(visible=False), gr.update(visible=False)