Update models/detection/detector.py

models/detection/detector.py

@@ -1,5 +1,5 @@
 import logging
-from PIL import Image
+from PIL import Image, ImageDraw
 from huggingface_hub import hf_hub_download
 from ultralytics import YOLO
 
@@ -48,7 +48,8 @@ class ObjectDetector:
         weights_path = hf_hub_download(
             repo_id=repo_id,
             filename=filename,
-            cache_dir="models/detection/weights"
+            cache_dir="models/detection/weights",
+            force_download=True #Clear cache
         )
 
         self.device = device