Update app.py
Browse files
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('
|
38 |
|
39 |
# Load the PCA instance
|
40 |
-
pca = load('
|
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
|