raffaelsiregar commited on
Commit
fc7fc7d
·
verified ·
1 Parent(s): c2f78e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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('best-model.pth', map_location=device)
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()