MoinulwithAI commited on
Commit
f9d8c0f
·
verified ·
1 Parent(s): 10a05ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ model.classifier = nn.Sequential(
17
  nn.Dropout(0.4),
18
  nn.Linear(model.classifier[1].in_features, 7) # num_classes = 7 (angry, disgust, fear, happy, neutral, sad, surprise)
19
  )
20
- model.load_state_dict(torch.load("best_mood_classifier.pth"))
21
  model = model.to(device)
22
  model.eval()
23
 
 
17
  nn.Dropout(0.4),
18
  nn.Linear(model.classifier[1].in_features, 7) # num_classes = 7 (angry, disgust, fear, happy, neutral, sad, surprise)
19
  )
20
+ model.load_state_dict(torch.load("best_mood_classifier.pth", map_location=torch.device('cpu')))
21
  model = model.to(device)
22
  model.eval()
23