Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -646,7 +646,7 @@ num_classes = len(dog_breeds)
|
|
646 |
model = BaseModel(num_classes=num_classes, device=device)
|
647 |
|
648 |
# Load model path
|
649 |
-
model_path =
|
650 |
checkpoint = torch.load(model_path, map_location=device)
|
651 |
|
652 |
# Load model state
|
|
|
646 |
model = BaseModel(num_classes=num_classes, device=device)
|
647 |
|
648 |
# Load model path
|
649 |
+
model_path = "124_best_model_dog.pth"
|
650 |
checkpoint = torch.load(model_path, map_location=device)
|
651 |
|
652 |
# Load model state
|