Update app.py

app.py

@@ -5,12 +5,20 @@ from transformers import pipeline
 classifier = pipeline(task="zero-shot-classification", model="facebook/bart-large-mnli")
 
 def analyze_sentiment(text):
-    results = classifier(text, ["Positivo", "Negativo", "Neutro"], multi_label=True)
+    results = classifier(text, ["positive", "negative", "neutral"], multi_label=True)
     mx = max(results['scores'])
     ind = results['scores'].index(mx)
-    result = results['labels'][ind]   
+    result = results['labels'][ind]
     
-    return result
+    # Traducción de etiquetas al español
+    if result == "positive":
+        sentiment = "Positivo"
+    elif result == "negative":
+        sentiment = "Negativo"
+    else:
+        sentiment = "Neutro"
+    
+    return sentiment
 
 # Configuración de la interfaz de Gradio
 demo = gr.Interface(