PruebaGradio / app.py
MarcoM003's picture
Create app.py
2dd6522 verified
raw
history blame
652 Bytes
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()