Update app.py

app.py

@@ -5,8 +5,11 @@ repo_id = "yejimene/clasificadorTexto"
 learner = from_pretrained_fastai(repo_id)
 
 def predict(text):
-    prediction_output = learner.predict(text)
-    print(f"Output de learner.predict('{text}'):")
-    print(prediction_output)  # Imprime la salida completa
-    return "Resultado impreso en la consola" 
+    pred,pred_idx,probs = learner.predict(text)
+    if (pred == "0"):
+        return "Negative"
+    elif (pred == "1"):
+        return "Positive"
+    else:
+        return "Error"
 gr.Interface(fn=predict, inputs="text", outputs="label", examples=["This movie was terrible and boring.","I loved this movie, it was amazing!"]).launch(share=False)