LearningnRunning's picture
FIX path error
3a53eae
raw
history blame
305 Bytes
import os
# 프로젝트 루트 디렉토리 경로 얻기
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
# MODEL_PATH를 절대 경로로 설정
DETECT_MODEL_PATH = os.path.join(BASE_DIR, 'weights', 'yolov9_c_nsfw.pt')
CLASSIFICATION_MODEL_PATH = "Falconsai/nsfw_image_detection"