File size: 195 Bytes
d2542a3 |
1 2 3 4 5 6 7 8 |
# config.py
import torch
UPLOAD_DIR = "backend/static/uploads"
OUTPUT_DIR = "backend/static/outputs"
FRAME_RATE = 15
SCORE_THRESHOLD = 0.4
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|