Spaces:

rgny
/

deneme

Runtime error

rgny commited on Jul 17, 2024

Commit

0af9b66

verified ·

1 Parent(s): b56a0a9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ effnetb2, effnetb2_transforms = create_model(
 # Load saved weights
 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
     )
 )

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