Update models/detection/detector.py

models/detection/detector.py

@@ -2,10 +2,12 @@ import logging
 from PIL import Image, ImageDraw
 from huggingface_hub import hf_hub_download
 from ultralytics import YOLO
+import shutil
 
 
-logger = logging.getLogger(__name__)
 
+logger = logging.getLogger(__name__)
+shutil.rmtree("models/detection/weights", ignore_errors=True)
 class ObjectDetector:
     """
     Generalized Object Detection Wrapper for YOLOv5, YOLOv8, and future variants.