Spaces:

remotewith
/

hat_final

Sleeping

remotewith commited on Jun 30, 2023

Commit

7dff8d2

1 Parent(s): d9c900c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

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