Update app.py
Browse files
app.py
CHANGED
@@ -68,7 +68,7 @@ class CNN1DLSTMAudioClassifier(nn.Module):
|
|
68 |
|
69 |
num_class = 6
|
70 |
device = torch.device('cpu')
|
71 |
-
state_dict = torch.load('
|
72 |
model = CNN1DLSTMAudioClassifier(num_class)
|
73 |
model.load_state_dict(state_dict)
|
74 |
model.eval()
|
|
|
68 |
|
69 |
num_class = 6
|
70 |
device = torch.device('cpu')
|
71 |
+
state_dict = torch.load('best_model.pth', map_location=device)
|
72 |
model = CNN1DLSTMAudioClassifier(num_class)
|
73 |
model.load_state_dict(state_dict)
|
74 |
model.eval()
|