import gradio as gr
from transformers import pipeline

# Cargar el modelo de Hugging Face
modelo = pipeline("sentiment-analysis")  # Cambia el tipo de modelo según tus necesidades

# Definir la función que se llamará al predecir
def predecir(texto):
    resultado = modelo(texto)
    return resultado[0]  # Retorna el primer resultado

# Crear la interfaz de Gradio
interfaz = gr.Interface(
    fn=predecir,
    inputs="text",
    outputs="json",
    title="Modelo de Análisis de Sentimiento",
    description="Escribe un texto para obtener su análisis de sentimiento."
)

# Iniciar la aplicación
if __name__ == "__main__":
    interfaz.launch()