John Sathya
Add RF-DETR detection service with FastAPI
202fa33
Raw
History Blame Contribute Delete
516 Bytes
import os
from pathlib import Path
# Model configuration - resolve to absolute path
_default_model = str(Path(__file__).resolve().parent.parent.parent / "checkpoint_best_ema.pth")
MODEL_PATH = os.getenv("MODEL_PATH", _default_model)
CONFIDENCE_THRESHOLD = float(os.getenv("CONFIDENCE_THRESHOLD", "0.3"))
# Server configuration
HOST = os.getenv("HOST", "0.0.0.0")
PORT = int(os.getenv("PORT", "8090"))
# Class names are auto-extracted from the checkpoint at startup.
# No need to hardcode them here.