import gradio as gr
from transformers import pipeline

# Cargar modelo desde Hugging Face
classifier = pipeline("text-classification", model="prabinpanta0/Patient-Readmission-Prediction")

# Función de predicción
def predict_readmission(text):
    result = classifier(text)
    return f"Probabilidad de readmisión: {result[0]['label']} (Score: {round(result[0]['score'], 2)})"

# Interfaz con Gradio
interface = gr.Interface(
    fn=predict_readmission,
    inputs=gr.Textbox(lines=4, placeholder="Ingresa la información del paciente..."),
    outputs="text",
    title="Predicción de Reingreso de Pacientes",
    description="Este modelo predice si un paciente será readmitido al hospital según su historial."
)

interface.launch()