Update app.py

app.py

@@ -19,7 +19,7 @@ def preprocess(text):
 def get_top_emojis(text, top_n=TOP_N):
     preprocessed = preprocess(text)
     inputs = tokenizer(preprocessed)
-    preds = model(**inputs)
+    preds = model(**inputs).logits
     scores = torch.nn.functional.softmax(preds, axis=-1).detach().numpy()
     ranking = np.argsort(scores)
     ranking = ranking[::-1][:top_n]