Update app.py

app.py

@@ -10,22 +10,22 @@ labels = learner.dls.vocab
 
 # Definimos una función que se encarga de llevar a cabo las predicciones
 def predict(txt):
-    output = learner.predict(txt)
-    print("Output of learner.predict:", output)  # Agrega esta línea para depuración
-    # Desempaquetamos basado en la estructura de output
-    if len(output) == 3:
-        pred, _, probs = output
-    elif len(output) == 2:
-        pred, probs = output
+    pred,_, probs = learner.predict(txt)
+    if pred == '0':
+        return "Positive"
+    else if pred == 1:
+        return "Neutral"
     else:
-        raise ValueError("Unexpected output format from learner.predict")
-    
-    return {labels[i]: float(probs[i]) for i in range(len(labels))}
+        return "Negative"
+
+        examp = ["Todo el mundo con la segunda temporada de Narcos y yo no, PORQUE NO TENGO NETFLIX @user pasaros una suscripcion de 5 dias",
+                "Me gusta estar sola y me molesta la gente que habla mucho. Si tan solo nos juntáramos a escuchar música y de rato en rato hablar de ello",
+                "La voz sepsi que debe tener Que dicha Gaby! Ya el lunes regresará "con toda la pata"]
 
 # Creamos la interfaz y la lanzamos. 
 gr.Interface(
     fn=predict, 
-    inputs=gr.inputs.Textbox(lines=2, placeholder="Ingresa el texto aquí..."), 
-    outputs=gr.outputs.Label(num_top_classes=3),
-    examples=[['Me siento muy feliz hoy'], ['Estoy triste'], ['Tengo miedo']]
+    inputs=gr.inputs.Textbox(lines=1, placeholder="Ingresa el texto aquí..."), 
+    outputs=gr.Textbox(label="Prediction"),
+    examples=examp
 ).launch(share=False)