Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -131,7 +131,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
|
| 131 |
model = BaseModel(num_classes=num_classes, device=device).to(device)
|
| 132 |
|
| 133 |
# Load model path
|
| 134 |
-
model_path = '
|
| 135 |
checkpoint = torch.load(model_path, map_location=device)
|
| 136 |
|
| 137 |
# Load model state
|
|
|
|
| 131 |
model = BaseModel(num_classes=num_classes, device=device).to(device)
|
| 132 |
|
| 133 |
# Load model path
|
| 134 |
+
model_path = '124_best_model_dog.pth'
|
| 135 |
checkpoint = torch.load(model_path, map_location=device)
|
| 136 |
|
| 137 |
# Load model state
|