saronium commited on
Commit
237e0fa
·
verified ·
1 Parent(s): c9f63b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -34,10 +34,10 @@ class ANNModel(nn.Module):
34
  ann_model = ANNModel()
35
 
36
  # Load the trained model
37
- ann_model.load_state_dict(torch.load('ann_model.pth'))
38
 
39
  # Load the PCA instance
40
- pca = load('pca.pkl')
41
 
42
  vgg16 = models.vgg16(pretrained=True).features
43
  # Function to load and preprocess a single audio file
 
34
  ann_model = ANNModel()
35
 
36
  # Load the trained model
37
+ ann_model.load_state_dict(torch.load('ann_model1.pth'))
38
 
39
  # Load the PCA instance
40
+ pca = load('pca1.pkl')
41
 
42
  vgg16 = models.vgg16(pretrained=True).features
43
  # Function to load and preprocess a single audio file