Spaces:

KillD00zer
/

fight-object_detection

Running

KillD00zer commited on Apr 18

Commit

b2f3c2c

verified ·

1 Parent(s): f9641b0

Update objec_detect_yolo.py

Files changed (1) hide show

objec_detect_yolo.py CHANGED Viewed

@@ -29,7 +29,7 @@ def detection(path: str) -> Tuple[Set[str], str]:
     # --- Model Loading ---
     # Construct path relative to this script file
-    model_path = os.path.join(os.path.dirname(__file__), "yolo", "best.pt")
     if not os.path.exists(model_path):
          raise FileNotFoundError(f"YOLO model file not found at: {model_path}")
     try:

     # --- Model Loading ---
     # Construct path relative to this script file
+    model_path = os.path.join(os.path.dirname(__file__), "best.pt")
     if not os.path.exists(model_path):
          raise FileNotFoundError(f"YOLO model file not found at: {model_path}")
     try: