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()]