Update app.py

app.py

@@ -14,19 +14,15 @@ def preprocess(text):
     return " ".join(new_text)
 
 
-def get_top_emojis(text, top_n=TOP_N):
+def get_top_emojis(text, tokenizer, top_n=TOP_N):
     preprocessed = preprocess(text)
     inputs = tokenizer(preprocessed, return_tensors="pt")
     preds = model(**inputs).logits
     scores = torch.nn.functional.softmax(preds, dim=-1).detach().numpy()
     ranking = np.argsort(scores)
-    print(ranking)
     ranking = ranking.squeeze()[::-1][:top_n]
-    print(scores)
-    print(ranking)
-    print(model.config.id2label)
     emojis = [model.config.id2label[i] for i in ranking]
-    return '/t'.join(map(str, emojis))
+    return '\t'.join(map(str, emojis))
 
 def main():
 
@@ -62,7 +58,7 @@ def main():
     # Define function to run when submit is clicked
     def submit(message):
         if len(message) > 0:
-            st.header(get_top_emojis(message))
+            st.header(get_top_emojis(message, tokenizer=tokenizer))
         else:
             st.error("The text can't be empty")