Salimshakeel's picture
optimize
6a6ee7b
raw
history blame
243 Bytes
# 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"