Update app.py

app.py

@@ -8,18 +8,26 @@ def load_model(selected_model):
     return loaded_model
 
 encoder = {
-        'negative':"a",
-        'neutral':"b",
-        'positive:"c"
-    }
+    'negative': "Negativo",
+    'neutral': "Neutro",
+    'positive': "Positivo"
+}
 
 classifier = pipeline(task="zero-shot-classification", model="facebook/bart-large-mnli")
+
 def analyze_sentiment(text):
-  results = classifier(text,["positive","negative",'neutral'],multi_label=True)
-  mx = max(results['scores'])
-  ind = results['scores'].index(mx)
-  result = results['labels'][ind]
-  return encoder[result]
+    results = classifier(text, ["positive", "negative", "neutral"], multi_label=True)
+    mx = max(results['scores'])
+    ind = results['scores'].index(mx)
+    result = results['labels'][ind]
+    return encoder[result]
+
+demo = gr.Interface(
+    fn=analyze_sentiment,
+    inputs="text",
+    outputs="text",
+    title="Análisis de Sentimientos"
+)
 
-demo = gr.Interface(fn=analyze_sentiment, inputs="text", outputs="text", title="Sentiment Analysis") 
 demo.launch(share=True)
+