|
from reader import get_article |
|
import gradio as gr |
|
from transformers import pipeline |
|
|
|
info = get_article() |
|
|
|
|
|
|
|
|
|
|
|
|
|
def classify_sentiment(audio): |
|
pipe = pipeline("audio-classification", model="hackathon-pln-es/wav2vec2-base-finetuned-sentiment-classification-MESD") |
|
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")] |
|
label = gr.outputs.Label(num_top_classes=5) |
|
|
|
|
|
|
|
description = """ Gradio demo for Sentiment Classification of Spanish audios using Wav2Vec2. This app can be powered by either of these two trained models : i) [Model A](https://huggingface.co./hackathon-pln-es/wav2vec2-base-finetuned-sentiment-classification-MESD) ii) [Model B](https://huggingface.co./hackathon-pln-es/wav2vec2-base-finetuned-sentiment-mesd) |
|
|
|
Note: The Audio examples provided for testing this app were randomly picked from the test dataset. |
|
""" |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
interface = gr.Interface(fn=classify_sentiment, inputs=input_audio, outputs=label, examples=[["basta_neutral.wav"], ["detras_disgust.wav"], ["mortal_sadness.wav"], ["respiracion_happiness.wav"], ["robo_fear.wav"]], article=info['article'], css=info['css'], theme='huggingface', title=info['title'], allow_flagging='never', description=description) |
|
interface.launch() |
|
|
|
|