adorkin commited on
Commit
29e5b04
·
1 Parent(s): a24a607

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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)