raffaelsiregar commited on
Commit
58d4def
·
verified ·
1 Parent(s): 1e5b0d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -69,7 +69,7 @@ class CNN1DLSTMAudioClassifier(nn.Module):
69
  num_class = 6
70
  model = CNN1DLSTMAudioClassifier(num_class)
71
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
72
- model = model.load_state_dict(torch.load("speech-emotion-recognition-best-model.bin", weights_only=False))
73
  model = model.to(device)
74
  model.eval()
75
 
 
69
  num_class = 6
70
  model = CNN1DLSTMAudioClassifier(num_class)
71
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
72
+ model = model.load_state_dict(torch.load("best-model-emotion-recognition.bin", weights_only=False))
73
  model = model.to(device)
74
  model.eval()
75