Update app.py
Browse files
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)
|