ReiderMx's picture
Create app.py
5981bca verified
raw
history blame
689 Bytes
import gradio as gr
from transformers import pipeline
# Configuraci贸n del clasificador de sentimientos
classifier = pipeline(task="zero-shot-classification", model="facebook/bart-large-mnli")
def analyze_sentiment(text):
results = classifier(text, ["Positivo", "Negativo", "Neutro"], multi_label=True)
mx = max(results['scores'])
ind = results['scores'].index(mx)
result = results['labels'][ind]
return result
# Configuraci贸n de la interfaz de Gradio
demo = gr.Interface(
fn=analyze_sentiment,
inputs="Escribe tu comentario",
outputs="Sentimiento",
title="An谩lisis de Sentimientos"
)
# Lanzamiento de la interfaz
demo.launch(share=True)