Update app.py
Browse files
app.py
CHANGED
@@ -23,6 +23,7 @@ def get_top_emojis(text, top_n=TOP_N):
|
|
23 |
preds = model(**inputs).logits
|
24 |
scores = torch.nn.functional.softmax(preds, dim=-1).detach().numpy()
|
25 |
ranking = np.argsort(scores)
|
|
|
26 |
ranking = ranking[::-1].squeeze()[:top_n]
|
27 |
print(scores)
|
28 |
print(ranking)
|
|
|
23 |
preds = model(**inputs).logits
|
24 |
scores = torch.nn.functional.softmax(preds, dim=-1).detach().numpy()
|
25 |
ranking = np.argsort(scores)
|
26 |
+
print(ranking)
|
27 |
ranking = ranking[::-1].squeeze()[:top_n]
|
28 |
print(scores)
|
29 |
print(ranking)
|