File size: 305 Bytes
a8f267e
 
 
 
 
3a53eae
 
1
2
3
4
5
6
7
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"