Jhoeel's picture
Create app.py
226cf35
raw
history blame
649 Bytes
# Importa librerias
from transformers import pipeline
import gradio as gr
# Inicializaci贸n de Modelo de Clasificaci贸n de Sentimiento
modelo = 'pysentimiento/robertuito-sentiment-analysis'
pipeline_de_clasificador = pipeline("text-classification",model=modelo)
def clasificador(texto):
"""
Funci贸n que clasifica un texto utilizando un pipeline de clasificaci贸n.
Par谩metros:
texto (str): El texto a clasificar.
Devuelve:
str: La clase asignada al texto.
"""
return str(pipeline_de_clasificador(texto)[0]['label'])
iface = gr.Interface(fn=clasificador, inputs="texto", outputs="texto")
iface.launch()