Spaces:

stefanT
/

SimpleClassifier

Runtime error

stefantaga commited on Apr 21, 2024

Commit

cff5998

1 Parent(s): 74c2e3b

final modification

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ effnetb2, effnetb2_transforms = create_effnetb2_model(
 effnetb2.load_state_dict(
     torch.load(
-        f="09_pretrained_effnetb2_feature_extractor_pizza_steak_sushi_20_percent.pth",
         map_location=torch.device("cpu"),  # load to CPU
     )
 )

 effnetb2.load_state_dict(
     torch.load(
+        f="07_effnetb2_data_20_percent_10_epochs.pth",
         map_location=torch.device("cpu"),  # load to CPU
     )
 )