Update app.py

app.py

@@ -1,32 +1,21 @@
 import gradio as gr
+import pickle
 from transformers import pipeline
 
-# Configuración del clasificador de sentimientos
-classifier = pipeline(task="zero-shot-classification", model="facebook/bart-large-mnli")
+def load_model(selected_model):
+    with open(selected_model, 'rb') as file:
+        loaded_model = pickle.load(file)
+    return loaded_model
+
 
-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]
-    
-    # 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(
-    fn=analyze_sentiment,
-    inputs="Escribe tu comentario",
-    outputs="Sentimiento", 
-    title="Análisis de Sentimientos"
-)
+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 result
 
-# Lanzamiento de la interfaz
+demo = gr.Interface(fn=analyze_sentiment, inputs="text", outputs="text", title="Sentiment Analysis") 
 demo.launch(share=True)