Spaces:

cruvss
/

Food_Classify

Build error

mulasagg commited on Nov 24, 2024

Commit

900b87d

1 Parent(s): c188c50

fix app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ effnetb2, effnetb2_transforms = create_effnetb2_model(
 #load saved weights
 effnetb2.load_state_dict(
     torch.load(
-        f="demos/foodvision_mini/Pretrained_effnet_v2_20_percent.pth,
         map_location=torch.device("cpu"),  # load to CPU
     )
 )

 #load saved weights
 effnetb2.load_state_dict(
     torch.load(
+        f="Pretrained_effnet_v2_20_percent.pth",#,"demos/foodvision_mini/Pretrained_effnet_v2_20_percent.pth",
         map_location=torch.device("cpu"),  # load to CPU
     )
 )