Update app.py

app.py

@@ -11,7 +11,7 @@ from joblib import load
 import soundfile as sf
 
 # Assuming you already have the 'ann_model' trained and 'pca' instance from the previous code
-language_mapping = {'malayalam': 0, 'english': 1, 'tamil': 2,'hindi':3,'kannada':4,'telugu':5}
+language_mapping = {'malayalam': 0, 'english': 1, 'tamil': 2,'hindi':3}
 
 class ANNModel(nn.Module):
     def __init__(self):
@@ -20,7 +20,7 @@ class ANNModel(nn.Module):
         self.relu1 = nn.ReLU()
         self.fc2 = nn.Linear(128, 64)
         self.relu2 = nn.ReLU()
-        self.fc3 = nn.Linear(64, 6)
+        self.fc3 = nn.Linear(64, 4)
 
     def forward(self, x):
         x = self.fc1(x)
@@ -34,10 +34,10 @@ class ANNModel(nn.Module):
 ann_model = ANNModel()
 
 # Load the trained model
-ann_model.load_state_dict(torch.load('ann_model.pth'))
+ann_model.load_state_dict(torch.load('ann_model1.pth'))
 
 # Load the PCA instance
-pca = load('pca.pkl')
+pca = load('pca1.pkl')
 
 vgg16 = models.vgg16(pretrained=True).features
 # Function to load and preprocess a single audio file