new model trained on larger dataset

app.py

@@ -8,7 +8,7 @@ token = "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"
 
 # Download model from HuggingFace Hub and load it with PyTorch
 REPO_ID = "gonzalocordova/DistractionDetectorCNN"
-FILENAME = "model_with_extended_dataset_resnet50_2023-04-20_11-24-17.pth"
+FILENAME = "resnet50_xlarge_resnet50_2023-05-19_21-41-02.pth"
 model_pth = hf_hub_download(repo_id=REPO_ID, filename=FILENAME, repo_type="model", use_auth_token=token)
 model = torch.load(model_pth, map_location=torch.device('cpu'))
 model.eval()