add show emotions name

app.py

@@ -14,7 +14,8 @@ def translate(text):
   logits = output.logits 
   prediction = F.softmax(logits, dim=1)
   y_pred = torch.argmax(prediction).numpy()
-  return y_pred
+  class_names = ['sad','joy','love','anger','fear','surprise']
+  return class_names[y_pred]
 #  output = model.generate(input["input_ids"], max_length=40, num_beams=4, early_stopping=True)
   
 title = "Text Emotion Classification"