Update app.py

app.py

@@ -3,6 +3,30 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import numpy as np
 import torch
 
+
+def preprocess(text):
+    new_text = []
+    for t in text.split(" "):
+        t = '@user' if t.startswith('@') and len(t) > 1 else t
+        t = 'http' if t.startswith('http') else t
+        new_text.append(t)
+    return " ".join(new_text)
+
+
+def get_top_emojis(text, 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))
+
 def main():
 
     st.set_page_config(  # Alternate names: setup_page, page, layout
@@ -34,31 +58,10 @@ def main():
     tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
     model = AutoModelForSequenceClassification.from_pretrained(BASE_MODEL)
 
-    def preprocess(text):
-        new_text = []
-        for t in text.split(" "):
-            t = '@user' if t.startswith('@') and len(t) > 1 else t
-            t = 'http' if t.startswith('http') else t
-            new_text.append(t)
-        return " ".join(new_text)
-
-    def get_top_emojis(text, 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 ', '.join(map(str, emojis))
 
     # Define function to run when submit is clicked
     def submit(message):
-        if len(message)>0:
+        if len(message) > 0:
             st.header(get_top_emojis(message))
         else:
             st.error("The text can't be empty")
@@ -68,8 +71,9 @@ def main():
         submit(message)
 
     st.text('')
-    st.markdown('<span style="color:blue; font-size:10px">App created by [@AlekseyDorkin](https://huggingface.co/AlekseyDorkin) \
-        and [@akshay7](https://huggingface.co/akshay7)</span>',unsafe_allow_html=True)
+    st.markdown('''<span style="color:blue; font-size:10px">App created by [@AlekseyDorkin](https://huggingface.co/AlekseyDorkin
+                and [@akshay7](https://huggingface.co/akshay7)</span>''', 
+                unsafe_allow_html=True)
 
 
 if __name__ == "__main__":