fix: setting torch device to cpu

app.py

@@ -10,7 +10,7 @@ token = "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"
 REPO_ID = "gonzalocordova/DistractionDetectorCNN"
 FILENAME = "model_with_extended_dataset_resnet50_2023-03-28_10-42-07.pth"
 model_pth = hf_hub_download(repo_id=REPO_ID, filename=FILENAME, repo_type="model", use_auth_token=token)
-model = torch.load(model_pth)
+model = torch.load(model_pth, map_location=torch.device('cpu'))
 model.eval()
 
 transform = transforms.Compose([