Hugging Face's logo

Spaces:
mladhd
/
DistractionDetector
Runtime error

App Files Community
gonzalocordova commited on
Commit
8d61ee9
·
1 Parent(s): 940281c

fix: specifying repo type

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from huggingface_hub import hf_hub_download
7
 
8
  REPO_ID = "gonzalocordova/DistractionDetectorCNN"
9
  FILENAME = "model_with_extended_dataset_resnet50_2023-03-28_10-42-07.pth"
10
- hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
11
  model = torch.load(os.path.join(os.getcwd(), FILENAME))
12
  model.eval()
13
 
 
7
 
8
  REPO_ID = "gonzalocordova/DistractionDetectorCNN"
9
  FILENAME = "model_with_extended_dataset_resnet50_2023-03-28_10-42-07.pth"
10
+ hf_hub_download(repo_id=REPO_ID, filename=FILENAME, repo_type="model")
11
  model = torch.load(os.path.join(os.getcwd(), FILENAME))
12
  model.eval()
13