Kumar
first commit
0077a91
raw
history blame
162 Bytes
import torch
MODEL_PATH = "xception_best_model.pth"
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
FRAME_SKIP = 5 # Every 5th frame