tzu commited on
Commit
00d3151
·
1 Parent(s): eb6ada7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def predict(speech):
14
  ds = load_dataset("hf-internal-testing/librispeech_asr_demo", "clean", split="validation")
15
  audio_file = ds[0]["audio"]["path"]
16
  audio_classifier = pipeline(
17
- task="audio-classification", model="ehcalabres/wav2vec2-lg-xlsr-en-speech-emotion-recognition"
18
  preds = audio_classifier(audio_file)
19
  preds = [{"score": round(pred["score"], 4), "label": pred["label"]} for pred in preds]
20
  return preds
 
14
  ds = load_dataset("hf-internal-testing/librispeech_asr_demo", "clean", split="validation")
15
  audio_file = ds[0]["audio"]["path"]
16
  audio_classifier = pipeline(
17
+ task="audio-classification", model="ehcalabres/wav2vec2-lg-xlsr-en-speech-emotion-recognition")
18
  preds = audio_classifier(audio_file)
19
  preds = [{"score": round(pred["score"], 4), "label": pred["label"]} for pred in preds]
20
  return preds