Update app.py

app.py

@@ -5,19 +5,38 @@ from torchvision import models
 from scipy.ndimage import zoom
 import gradio as gr
 import pickle
+from joblib import load
 
 
 # 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}
 
+class ANNModel(nn.Module):
+    def __init__(self):
+        super(ANNModel, self).__init__()
+        self.fc1 = nn.Linear(300, 128)
+        self.relu1 = nn.ReLU()
+        self.fc2 = nn.Linear(128, 64)
+        self.relu2 = nn.ReLU()
+        self.fc3 = nn.Linear(64, 6)
+
+    def forward(self, x):
+        x = self.fc1(x)
+        x = self.relu1(x)
+        x = self.fc2(x)
+        x = self.relu2(x)
+        x = self.fc3(x)
+        return x
+
+# Create an instance of your model
+ann_model = ANNModel()
 
 # Load the trained model
-with open('ann_model.pth', 'rb') as f:
-    ann_model = torch.load(f)
+ann_model.load_state_dict(torch.load('ann_model.pth'))
 
 # Load the PCA instance
-with open('pca.pkl', 'rb') as f:
-    pca = pickle.load(f)
+pca = load('pca.pkl')
+
 vgg16 = models.vgg16(pretrained=True).features
 # Function to load and preprocess a single audio file
 def preprocess_single_audio_vgg16(audio_file, vgg16_model, pca_instance):