Update app.py

app.py

@@ -24,13 +24,11 @@ def get_top_emojis(text, top_n=TOP_N):
     scores = torch.nn.functional.softmax(preds, dim=-1).detach().numpy()
     ranking = np.argsort(scores)
     print(ranking)
-    ranking = ranking[::-1].squeeze()[:top_n]
+    ranking = ranking.squeeze()[::-1][:top_n]
     print(scores)
     print(ranking)
     print(model.config.id2label)
-    for i in ranking:
-        print(model.config.id2label[i])
-    emojis = []
+    emojis = [model.config.id2label[i] for i in ranking]
     return emojis