Update app.py

app.py

@@ -7,6 +7,22 @@ repo_id = "ancebuc/entregable3"
 learner = from_pretrained_fastai(repo_id)
 labels = [i for i in range(6)]
 
+def postprocessing(pred):
+    if pred[0] == '0':
+        return json.dumps({"label": "Sadness"})
+    elif pred[0] == '1':
+        return json.dumps({"label": "Joy"})
+    elif pred[0] == '2':
+        return json.dumps({"label": "Love"})
+    elif pred[0] == '3':
+        return json.dumps({"label": "Anger"})
+    elif pred[0] == '4':
+        return json.dumps({"label": "Fear"})
+    elif pred[0] == '5':
+        return json.dumps({"label": "Surprise"})
+    else:
+        return json.dumps({"label": "Don't know"})
+        
 # Definimos una función que se encarga de llevar a cabo las predicciones
 def predict(text):
     #img = PILImage.create(img)
@@ -14,4 +30,4 @@ def predict(text):
     return pred[0]
     
 # Creamos la interfaz y la lanzamos. 
-gr.Interface(fn=predict, inputs=gr.inputs.Textbox(), outputs=gr.outputs.Label()).launch(share=False)
+gr.Interface(fn=predict, inputs=gr.inputs.Textbox(), outputs=gr.outputs.Label(preprocessing_fn=preprocessing)).launch(share=False)