Spaces:

AliKhalaji
/

FoodVision

Sleeping

AliKhalaji commited on May 29, 2025

Commit

9f2afdc

verified ·

1 Parent(s): 17d7633

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ effnetb2, effnetb2_transforms = create_effnetb2_model(
 effnetb2.load_state_dict(
     torch.load(
-        f="09_pretrained_effnetb2_feature_extractor_food101_20_percent.pth",
         map_location=torch.device("cpu"),
     )
 )

 effnetb2.load_state_dict(
     torch.load(
+        f="pretrained_effnetb2_feature_extractor_food101_20_percent.pth",
         map_location=torch.device("cpu"),
     )
 )