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