Spaces:

umasiyer
/

south-indian-breakfast

Sleeping

umasiyer commited on Jul 9

Commit

32f1713

•

1 Parent(s): 576d9cf

updated name issue

Files changed (1) hide show

app.py CHANGED Viewed

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

 # Load saved weights
 effnetb2.load_state_dict(
     torch.load(
+        f="indi-bf-classifier.pth",
         map_location=torch.device("cpu"),  # load to CPU
     )
 )