ADD: caching of models 2

app.py

@@ -6,10 +6,6 @@ import torch
 TOP_N = 5
 
 def main():
-    print("cur_model", cur_model_name)
-    
-    tokenizer = AutoTokenizer.from_pretrained(cur_model_name)
-    model = AutoModelForSequenceClassification.from_pretrained(cur_model_name)
     
     st.set_page_config(  # Alternate names: setup_page, page, layout
         layout="centered",  # Can be "centered" or "wide". In the future also "dashboard", etc.
@@ -77,7 +73,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, model=model))
+            st.header(get_top_emojis(message))
         else:
             st.error("The text can't be empty")