Update app.py

app.py

@@ -21,7 +21,7 @@ def get_top_emojis(text, top_n=TOP_N):
     preprocessed = preprocess(text)
     inputs = tokenizer(preprocessed, return_tensors="pt")
     preds = model(**inputs).logits
-    scores = torch.nn.functional.softmax(preds, axis=-1).detach().numpy()
+    scores = torch.nn.functional.softmax(preds, dim=-1).detach().numpy()
     ranking = np.argsort(scores)
     ranking = ranking[::-1][:top_n]
     emojis = [model.config.id2label[index] for index in ranking]