Spaces:
Runtime error
Runtime error
Commit ·
940281c
1
Parent(s): 57a5006
fix bug
Browse files
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
|
| 11 |
-
model = torch.load(os.path.join(os.getcwd(),
|
| 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([
|