remove test existing model

config.yaml

@@ -1,6 +1,6 @@
 model:
-  path: 'nezahatkorkmaz/traffic-sign-detection' # YOLOv8 traffic sign model from HuggingFace (better trained)
-  confidence_threshold: 0.35 # Minimum confidence for detections
+  path: 'VietCat/GTSRB-Model/models/GTSRB.pt' # Path to the YOLO model on Hugging Face Hub (will be downloaded automatically)
+  confidence_threshold: 0.15 # Minimum confidence for detections
 
 inference:
   box_color: (128, 0, 128) # Purple color for bounding boxes (BGR format)

model.py

@@ -22,24 +22,18 @@ class TrafficSignDetector:
             # Load model from path
             model_path = config['model']['path']
             
-            # If it's a HuggingFace path, use hf_hub_download
-            if '/' in model_path and not model_path.endswith('.pt'):
-                # HuggingFace model repo
-                parts = model_path.split('/')
-                if len(parts) == 2:
-                    repo_id = model_path
-                    # Try to download best.pt or runs/detect/.../weights/best.pt
-                    try:
-                        local_model_path = hf_hub_download(repo_id=repo_id, filename='best.pt')
-                    except:
-                        local_model_path = hf_hub_download(repo_id=repo_id, filename='runs/detect/train/weights/best.pt')
-                    self.model = YOLO(local_model_path)
-                else:
-                    # Full path with filename
-                    repo_id = '/'.join(parts[:-1])
-                    file_path = parts[-1]
+            # Handle HuggingFace paths
+            if '/' in model_path:
+                if model_path.endswith('.pt'):
+                    # Full path with filename (e.g., VietCat/GTSRB-Model/models/GTSRB.pt)
+                    parts = model_path.rsplit('/', 1)
+                    repo_id = parts[0]
+                    file_path = parts[1]
                     local_model_path = hf_hub_download(repo_id=repo_id, filename=file_path)
                     self.model = YOLO(local_model_path)
+                else:
+                    # Just repo name (e.g., user/model-name)
+                    self.model = YOLO(model_path)
             else:
                 # Local path or direct model path
                 self.model = YOLO(model_path)