Update app.py

app.py

@@ -1,11 +1,31 @@
 import gradio as gr
-from setfit import SetFitModel
+import os
 
-# Cargar el modelo
-model = SetFitModel.from_pretrained("rovargasc/modelopruebaUNAL2")
+try:
+    from setfit import SetFitModel
+    print("✓ SetFit importado correctamente")
+except ImportError as e:
+    print(f"❌ Error importando SetFit: {e}")
+    raise e
+
+# Cargar el modelo con manejo de errores
+def load_model():
+    try:
+        print("Cargando modelo...")
+        model = SetFitModel.from_pretrained("rovargasc/modelopruebaUNAL2")
+        print("✓ Modelo cargado correctamente")
+        return model
+    except Exception as e:
+        print(f"❌ Error cargando el modelo: {e}")
+        return None
+
+model = load_model()
 
 def predict_text(text):
     """Función para hacer predicciones con el modelo"""
+    if model is None:
+        return "Error: El modelo no se pudo cargar"
+    
     if not text.strip():
         return "Por favor ingresa un texto"
     
@@ -13,7 +33,7 @@ def predict_text(text):
         prediction = model(text)
         return f"Predicción: {prediction}"
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"Error en la predicción: {str(e)}"
 
 # Crear la interfaz
 interface = gr.Interface(