gonzalocordova commited on
Commit
940281c
·
1 Parent(s): 57a5006
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,8 @@ 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, output_dir=os.path.join(os.getcwd(), "model"))
11
- model = torch.load(os.path.join(os.getcwd(), "model"))
12
  model.eval()
13
 
14
  transform = transforms.Compose([
 
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
 
14
  transform = transforms.Compose([