Update app.py

app.py

@@ -5,24 +5,9 @@ from fastai.vision.all import *
 repo_id = "ancebuc/entregable3"
 
 learner = from_pretrained_fastai(repo_id)
-labels = [i for i in range(6)]
+labels = ['Sadness', 'Joy', 'Love', 'Anger', 'Fear', 'Surprise']
 
-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)
@@ -30,5 +15,4 @@ def predict(text):
     return pred 
     
 # Creamos la interfaz y la lanzamos. 
-label = gr.outputs.Label(postprocessing_fn=postprocessing)
-gr.Interface(fn=predict, inputs=gr.inputs.Textbox(), outputs=label).launch(share=False)
+gr.Interface(fn=predict, inputs=gr.inputs.Textbox(), outputs=gr.outputs.Label()).launch(share=False)