Update app.py

app.py

@@ -10,7 +10,12 @@ labels = learner.dls.vocab
 # Definimos una función que se encarga de llevar a cabo las predicciones
 def predict(text):
     pred, pred_idx, probs = learner.predict(text)
-    return pred[0]
+    if pred[0]==0:
+        return "negative"
+    elif pred[0]==1:
+        return "neutral"
+    else:
+        return "positive"
 
 # Creamos la interfaz y la lanzamos.
-gr.Interface(fn=predict, inputs="text", outputs="text", title="Clasificador de Textos Económicos", description="Introduce un texto y obtén la clase a la que pertenece.").launch(share=False)
+gr.Interface(fn=predict, inputs="text", outputs="text", title="Clasificador de Textos Económicos", description="Introduce un texto en inglés y obtén la clase a la que pertenece.").launch(share=False)