Update app.py

app.py

@@ -8,8 +8,7 @@ import torch
 # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
 repo_id = "islasher/clasificador-dair-emotion"
 
-# learner = from_pretrained_fastai(repo_id)
-# labels = learner.dls.vocab
+labels = ['sadness', 'joy', 'love', 'anger', 'fear', 'surprise']
 
 # Definimos una función que se encarga de llevar a cabo las predicciones
 
@@ -25,7 +24,7 @@ def predict(frase):
     inputs = tokenizer(frase, return_tensors="pt")
     outputs = model(**inputs)
     predicted_class = torch.argmax(outputs.logits, dim=1).item()
-    return predicted_class
+    return labels[predicted_class]
     
 # Creamos la interfaz y la lanzamos. 
 gr.Interface(fn=predict, inputs="text", outputs="text").launch(share=False)