DawnC commited on
Commit
c88e383
1 Parent(s): cb218a5

Update app.py

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