Update app.py
Browse files
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("
|
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 |
|