Spaces:

Sushrut98
/

emotion

Sleeping

Sushrut98 commited on Feb 21, 2024

Commit

790e8db

verified ·

1 Parent(s): 5ad3e9e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,7 +32,7 @@ transform = transforms.Compose([
 # Load saved weights
 TinyCNN_model.load_state_dict(
     torch.load(
-        f="TinyCNN.pth",
         map_location=torch.device("cpu"),  # load to CPU
     )
 )

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