LearningnRunning's picture
feat Only detail detect model
82654de
raw
history blame contribute delete
309 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"