rovargasc/Setfit_Test

1import gradio as gr2import os3 4try:5    from setfit import SetFitModel6    print("✓ SetFit importado correctamente")7except ImportError as e:8    print(f"❌ Error importando SetFit: {e}")9    raise e10 11# Cargar el modelo con manejo de errores12def load_model():13    try:14        print("Cargando modelo...")15        model = SetFitModel.from_pretrained("rovargasc/modelopruebaUNAL2")16        print("✓ Modelo cargado correctamente")17        return model18    except Exception as e:19        print(f"❌ Error cargando el modelo: {e}")20        return None21 22model = load_model()23 24def predict_text(text):25    """Función para hacer predicciones con el modelo"""26    if model is None:27        return "Error: El modelo no se pudo cargar"28    29    if not text.strip():30        return "Por favor ingresa un texto"31    32    try:33        prediction = model(text)34        return f"Predicción: {prediction}"35    except Exception as e:36        return f"Error en la predicción: {str(e)}"37 38# Crear la interfaz39interface = gr.Interface(40    fn=predict_text,41    inputs=gr.Textbox(42        lines=3, 43        placeholder="Escribe tu texto aquí...",44        label="Texto de entrada"45    ),46    outputs=gr.Textbox(label="Resultado"),47    title="Clasificador de Texto - SetFit Model",48    description="Ingresa un texto para obtener una predicción del modelo SetFit",49    examples=[50        ["This movie is amazing!"],51        ["I don't like this product"],52        ["chokes on its own depiction of upper-crust decorum."]53    ]54)55 56if __name__ == "__main__":57    interface.launch()