Update app.py

app.py

@@ -17,10 +17,11 @@ def get_emotion(microphone, file_upload, task):
     elif (microphone is None) and (file_upload is None):
         return "ERROR: You have to either use the microphone or upload an audio file"
 
-    
+    model = torch.load('model.pth')
     file = microphone if microphone is not None else file_upload
-    print(file)
-    test = feature_extractor(file, sampling_rate=16000, padding=True, return_tensors="pt" )       
+    speech, _ = librosa.load(file, sr=16000, mono=True)
+    test = feature_extractor(speech, sampling_rate=16000, padding=True, return_tensors="pt" )       
+    
     logits = model(**test).logits
     predicted_ids = torch.argmax(logits, dim=-1)
     labels = [model.config.id2label[_id] for _id in predicated_ids.tolist()]