Spaces:

Shriharsh
/

Food_Vision_101

Runtime error

Shriharsh commited on Jun 15, 2023

Commit

fae1c18

1 Parent(s): fe7b068

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ effnetb2, effnetb2_transforms = create_effnetb2_model(num_classes=101)
 # Load saved weights
 effnetb2.load_state_dict(
-    torch.load(f="09_pretrained_effnetb2_feature_extractor_food101_20_percent.pth",
                map_location=torch.device("cpu")) # load to CPU
 )

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