Update app.py

app.py

@@ -2,15 +2,15 @@
 import gradio as gr
 from transformers import pipeline
 
-#Model_1 = "hackathon-pln-es/wav2vec2-base-finetuned-sentiment-classification-MESD"
-#Model_2 ="hackathon-pln-es/wav2vec2-base-finetuned-sentiment-mesd"
+Model_1 = "hackathon-pln-es/wav2vec2-base-finetuned-sentiment-classification-MESD"
+Model_2 ="hackathon-pln-es/wav2vec2-base-finetuned-sentiment-mesd"
 
 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"), #gr.inputs.Dropdown([Model_1, Model_2], label="Model Name")]
+input_audio = [gr.inputs.Audio(source="microphone", type="filepath", label="Record/ Drop audio"), gr.inputs.Dropdown([Model_1, Model_2], label="Model Name")]
 label = gr.outputs.Label(num_top_classes=5)
 
 ################### Gradio Web APP ################################
@@ -36,6 +36,6 @@ 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"]],
+    examples=[["basta_neutral.wav", Model_1], ["detras_disgust.wav", Model_1], ["mortal_sadness.wav", Model_1], ["respiracion_happiness.wav", Modle_1], ["robo_fear.wav", Model_1]],
     theme="grass").launch()