Update app.py

app.py

@@ -14,7 +14,7 @@ def preprocess(text):
     return " ".join(new_text)
 
 
-def get_top_emojis(text, tokenizer, top_n=TOP_N):
+def get_top_emojis(text, tokenizer, model, top_n=TOP_N):
     preprocessed = preprocess(text)
     inputs = tokenizer(preprocessed, return_tensors="pt")
     preds = model(**inputs).logits
@@ -58,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, tokenizer=tokenizer))
+            st.header(get_top_emojis(message, tokenizer=tokenizer, model=model))
         else:
             st.error("The text can't be empty")