Update app.py

app.py

@@ -1,18 +1,17 @@
-
 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_name2id = {"Model A": "hackathon-pln-es/wav2vec2-base-finetuned-sentiment-classification-MESD", "Model B": "hackathon-pln-es/wav2vec2-base-finetuned-sentiment-mesd"}
+#model_name2id = {"Model A": "hackathon-pln-es/wav2vec2-base-finetuned-sentiment-classification-MESD", "Model B": "hackathon-pln-es/wav2vec2-base-finetuned-sentiment-mesd"}
 
-def classify_sentiment(audio, model_name):
-  pipe = pipeline("audio-classification", model=model_name2id[model_name])
+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_name2id["Model A"], model_name2id["Model B"], label="Model Name")]
+input_audio = [gr.inputs.Audio(source="microphone", type="filepath", label="Record/ Drop audio")]
 label = gr.outputs.Label(num_top_classes=5)
 
 ################### Gradio Web APP ################################
@@ -38,6 +37,5 @@ gr.Interface(
     fn = classify_sentiment,
     inputs = input_audio,
     outputs = label,
-    examples=[["basta_neutral.wav", model_name2id], ["detras_disgust.wav", model_name2id], ["mortal_sadness.wav", model_name2id], ["respiracion_happiness.wav", model_name2id], ["robo_fear.wav", model_name2id]],
+    examples=[["basta_neutral.wav"], ["detras_disgust.wav"], ["mortal_sadness.wav"], ["respiracion_happiness.wav"], ["robo_fear.wav"]],
     theme="grass").launch()
-