Skip to content

Commit 982fb81

Browse files
committed
update model path with huggingface
1 parent 397473e commit 982fb81

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

core/load_model.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,9 @@
44
import torch
55
import torch.nn.functional as F
66
from .config import DEVICE, MODEL_NAME, PRETRAINED, CLOTH_LABELS, RECORD_LABELS ,PROMPTS
7+
from huggingface_hub import hf_hub_download
78

9+
model_path = hf_hub_download(repo_id="swwho/Fashion-YOLO", filename="fashion_2.pt")
810

911
# --- 전역 변수 (Global State) ---
1012
GLOBAL_MODEL = None
@@ -75,7 +77,7 @@ def load_clip_model():
7577

7678

7779

78-
def load_yolo_model(model_path: str = 'c://Users/swwho/Desktop/codive/models/fashion.pt'):
80+
def load_yolo_model(model_path: str = model_path):
7981
"""
8082
YOLO 모델을 로드 함수
8183
"""

0 commit comments

Comments
 (0)