Hugging Face's logo

Spaces:
mladhd
/
DistractionDetector
Runtime error

App Files Community
gonzalocordova commited on
Commit
5f6bc53
·
1 Parent(s): 88ac1ec

feat: new model (change HF path)

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ token = "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"
8
 
9
  # Download model from HuggingFace Hub and load it with PyTorch
10
  REPO_ID = "gonzalocordova/DistractionDetectorCNN"
11
- FILENAME = "model_with_extended_dataset_resnet50_2023-03-28_10-42-07.pth"
12
  model_pth = hf_hub_download(repo_id=REPO_ID, filename=FILENAME, repo_type="model", use_auth_token=token)
13
  model = torch.load(model_pth, map_location=torch.device('cpu'))
14
  model.eval()
 
8
 
9
  # Download model from HuggingFace Hub and load it with PyTorch
10
  REPO_ID = "gonzalocordova/DistractionDetectorCNN"
11
+ FILENAME = "model_with_extended_dataset_resnet50_2023-04-20_11-24-17.pth"
12
  model_pth = hf_hub_download(repo_id=REPO_ID, filename=FILENAME, repo_type="model", use_auth_token=token)
13
  model = torch.load(model_pth, map_location=torch.device('cpu'))
14
  model.eval()