DavSem94 commited on
Commit
7cd26db
1 Parent(s): 4ce6817

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ effnetb2, effnetb2_transforms = create_effnetb2_model()
18
 
19
  # Load save weights
20
  effnetb2.load_state_dict(
21
- torch.load(data = "09_pretrained_effnetb2_feature_extractor_pizza_steak_sushi_20_percent.pth",
22
  map_location = torch.device("cpu"))) # Load the model to the CPU))
23
 
24
  ### 3. Predict function ###
 
18
 
19
  # Load save weights
20
  effnetb2.load_state_dict(
21
+ torch.load(f = "09_pretrained_effnetb2_feature_extractor_pizza_steak_sushi_20_percent.pth",
22
  map_location = torch.device("cpu"))) # Load the model to the CPU))
23
 
24
  ### 3. Predict function ###