DawnC commited on
Commit
1dc35ad
1 Parent(s): ca056ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = '124_82.30_best_model_dog.pth'
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