Spaces:

somosnlp-hackathon-2022
/

Audio-Sentiment-Classifier

Running

DrishtiSharma commited on Mar 30, 2022

Commit

f8d2495

•

1 Parent(s): d6a0552

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ def classify_sentiment(audio, model):
   pred = pipe(audio)
   return {dic["label"]: dic["score"] for dic in pred}
-input_audio = [gr.inputs.Audio(source="microphone", type="filepath", label="Record/ Drop audio"), gr.inputs.Dropdown(["DrishtiSharma/wav2vec2-base-finetuned-sentiment-mesd-v11", "hackathon-pln-es/wav2vec2-base-finetuned-sentiment-mesd"], label="Model Name")]
 label = gr.outputs.Label(num_top_classes=5)
 ################### Gradio Web APP ################################
@@ -33,6 +33,6 @@ gr.Interface(
     fn = classify_sentiment,
     inputs = input_audio,
     outputs = label,
-    #examples=[["test1.wav", "DrishtiSharma/wav2vec2-base-finetuned-sentiment-mesd-v11"], ["test2.wav", "hackathon-pln-es/wav2vec2-base-finetuned-sentiment-mesd"]],
     theme="grass", description=description).launch()

   pred = pipe(audio)
   return {dic["label"]: dic["score"] for dic in pred}
+input_audio = [gr.inputs.Audio(source="microphone", type="filepath", label="Record/ Drop audio"), gr.inputs.Dropdown(["hackathon-pln-es/wav2vec2-base-finetuned-sentiment-classification-MESD", "hackathon-pln-es/wav2vec2-base-finetuned-sentiment-mesd"], label="Model Name")]
 label = gr.outputs.Label(num_top_classes=5)
 ################### Gradio Web APP ################################
     fn = classify_sentiment,
     inputs = input_audio,
     outputs = label,
+    #examples=[["test1.wav", "hackathon-pln-es/wav2vec2-base-finetuned-sentiment-classification-MESD"], ["test2.wav", "hackathon-pln-es/wav2vec2-base-finetuned-sentiment-mesd"]],
     theme="grass", description=description).launch()