# 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" | |
# 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" | |