Update app.py

app.py

@@ -14,16 +14,16 @@ emotion_labels = [
 # Carga del modelo desde HuggingFace Hub
 learn = from_pretrained_fastai("diribes/go_emotions")
 
-# Función de predicción
 def clasificar_texto(texto):
-    pred, idxs, probs = learn.predict(texto)
+    pred, _, probs = learn.predict(texto)
 
-    # Si el modelo es multietiqueta (lista de índices)
-    if isinstance(pred, list):
-        pred_labels = [emotion_labels[int(i)] for i in pred]
-        return ", ".join(pred_labels)
-    
-    # Si es una sola etiqueta
+    # Si es MultiCategory (multietiqueta), conviértelo a lista de strings
+    if hasattr(pred, 'items'):
+        indices_predichos = [i for i, val in enumerate(pred) if val == 1]
+        emociones = [emotion_labels[i] for i in indices_predichos]
+        return ", ".join(emociones) if emociones else "Sin emoción detectada"
+
+    # Si es una única categoría (clasificación simple)
     return emotion_labels[int(pred)]
 
 # Interfaz Gradio