aayanb09 commited on
Commit
aec5a02
·
verified ·
1 Parent(s): 678ba37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ print(f"Using device: {device}")
15
  model = models.mobilenet_v2(pretrained=False)
16
  num_ftrs = model.classifier[1].in_features
17
  model.classifier[1] = nn.Linear(num_ftrs, len(class_names))
18
- model.load_state_dict(torch.load("ingredientRecognitionModel.pth", map_location=device))
19
  model = model.to(device)
20
  model.eval()
21
 
 
15
  model = models.mobilenet_v2(pretrained=False)
16
  num_ftrs = model.classifier[1].in_features
17
  model.classifier[1] = nn.Linear(num_ftrs, len(class_names))
18
+ model.load_state_dict(torch.load("ingredient_recognition_model.pth", map_location=device))
19
  model = model.to(device)
20
  model.eval()
21