Update app.py

app.py

@@ -12,9 +12,22 @@ import time
 import tempfile
 #test
 # --- Configuration ---
+HF_TOKEN = os.environ.get("HF_TOKEN")
+
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
-YOLO_PERSON_MODEL_PATH = 'yolov8n.pt'  # Standard YOLOv8 for person detection
-YOLO_FASHION_MODEL_PATH = 'best.pt'    # <<< Your custom fashion model path
+YOLO_PERSON_MODEL_PATH = hf_hub_download(
+    repo_id="dzmu/dripai-models",
+    filename="yolov8n.pt",
+    token=HF_TOKEN,
+    repo_type="model"
+)
+
+YOLO_FASHION_MODEL_PATH = hf_hub_download(
+    repo_id="dzmu/dripai-models",
+    filename="best.pt",
+    token=HF_TOKEN,
+    repo_type="model"
+)
 CLIP_MODEL_NAME = "ViT-B/32"
 
 # Confidence Thresholds