Update app.py

app.py

@@ -14,7 +14,7 @@ labels = ['Mundo', 'Deportes', 'Negocios', 'Ciencia/Tecnología']
 # Definir una función para predecir la categoría de un texto dado
 def predict(text):
     # Obtener las probabilidades de las etiquetas desde el modelo
-    probs = learner.predict(text)
+    probs = learner.predict(text)[-1]
     # Devolver un diccionario que mapea cada etiqueta a su probabilidad correspondiente
     return {labels[i]: float(probs[i]) for i in range(len(labels))}