app.py

import gradio as gr
from setfit import SetFitModel

# Cargar el modelo
model = SetFitModel.from_pretrained("desarrolloasesoreslocales/SetFitPruebaRecorte")

# Mapear las etiquetas

# Definir la función de predicción
def predict(payload):
    results = model.predict_proba([payload])[0]
    
    # return "\n".join([f"Label: {id2label[i]},      Score: {round(result.item()*100, 2)} %" for i, result in enumerate(results)])
    return results

# Crear una interfaz Gradio
iface = gr.Interface(
    fn=predict, 
    inputs=gr.inputs.Textbox(), 
    outputs=gr.outputs.Textbox(), 
    live=False,
    layout='vertical',
    title="Clasificador de Texto"
)

# Iniciar la interfaz Gradio
iface.launch()