Spaces:
Runtime error
Runtime error
Commit
·
314aa8a
1
Parent(s):
ec4daae
Update app.py
Browse files
app.py
CHANGED
@@ -5,8 +5,8 @@ from transformers import pipeline
|
|
5 |
|
6 |
def classify_sentiment(audio, model):
|
7 |
pipe = pipeline("audio-classification", model=model)
|
8 |
-
sentiment_classifier = pipe(audio)
|
9 |
-
return sentiment_classifier
|
10 |
|
11 |
|
12 |
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")]
|
|
|
5 |
|
6 |
def classify_sentiment(audio, model):
|
7 |
pipe = pipeline("audio-classification", model=model)
|
8 |
+
sentiment_classifier = pipe(audio)[0]
|
9 |
+
return sentiment_classifier
|
10 |
|
11 |
|
12 |
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")]
|