adorkin commited on
Commit
e128c45
1 Parent(s): 3bf5e3c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def preprocess(text):
18
 
19
  def get_top_emojis(text, top_n=TOP_N):
20
  preprocessed = preprocess(text)
21
- inputs = tokenizer(preprocessed)
22
  preds = model(**inputs).logits
23
  scores = torch.nn.functional.softmax(preds, axis=-1).detach().numpy()
24
  ranking = np.argsort(scores)
 
18
 
19
  def get_top_emojis(text, top_n=TOP_N):
20
  preprocessed = preprocess(text)
21
+ inputs = tokenizer(preprocessed, return_tensors="pt")
22
  preds = model(**inputs).logits
23
  scores = torch.nn.functional.softmax(preds, axis=-1).detach().numpy()
24
  ranking = np.argsort(scores)