DrishtiSharma commited on
Commit
347a04f
1 Parent(s): c72c353

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,9 +1,10 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- pipe = pipeline("audio-classification", model="hackathon-pln-es/wav2vec2-base-finetuned-sentiment-mesd")
5
 
6
- def classify_sentiment(audio):
 
 
7
  sentiment_classifier = pipe(audio)
8
  return sentiment_classifier
9
 
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
 
4
 
5
+
6
+ def classify_sentiment(audio, model):
7
+ pipe = pipeline("audio-classification", model=model)
8
  sentiment_classifier = pipe(audio)
9
  return sentiment_classifier
10