state farm dataset

.gitignore

@@ -1,7 +1,5 @@
 # Datasets - exclude all except samples and state-farm
 datasets/
-!datasets/text/val/
-!datasets/state-farm/
 
 # Environment
 .env

app/models/state_farm_model.py

@@ -40,13 +40,15 @@ class StateFarmDetector:
             return
         
         try:
-            # Load model directly from HuggingFace (YOLO handles this)
-            # If token is needed, YOLO will use environment variable or we can pass it
-            if HF_TOKEN:
-                # Set token as environment variable for YOLO to use
-                os.environ['HF_TOKEN'] = HF_TOKEN
+            # Download model file from HuggingFace
+            model_path = hf_hub_download(
+                repo_id=STATE_FARM_MODEL,
+                filename="weights/best.pt",
+                token=HF_TOKEN
+            )
             
-            self.model = YOLO(STATE_FARM_MODEL)
+            # Load YOLO model from downloaded file
+            self.model = YOLO(model_path)
             print(f"State Farm detection model loaded successfully from {STATE_FARM_MODEL}")
             
         except Exception as e: