File size: 243 Bytes
d2542a3
 
6a6ee7b
 
 
d2542a3
 
 
1
2
3
4
5
6
7
8
9
# config.py
import torch
import os
UPLOAD_DIR = os.path.join(os.getcwd(), "static/uploads")
OUTPUT_DIR = os.path.join(os.getcwd(), "static/outputs")
FRAME_RATE = 15
SCORE_THRESHOLD = 0.4
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"