Spaces:

mrdbourke
/

foodvision_big

Runtime error

mrdbourke commited on Aug 11, 2022

Commit

ec3bf01

1 Parent(s): a80b887

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ model, transforms = create_effnetb2_model(
 # Load saved weights
 model.load_state_dict(
     torch.load(
-        f= foodvision_big_demo_path / "09_pretrained_effnetb2_feature_extractor_food101_20_percent.pth",
         map_location=torch.device("cpu"),  # load to CPU
     )
 )

 # Load saved weights
 model.load_state_dict(
     torch.load(
+        f="09_pretrained_effnetb2_feature_extractor_food101_20_percent.pth",
         map_location=torch.device("cpu"),  # load to CPU
     )
 )