Update app.py

app.py

@@ -4,6 +4,7 @@ import numpy as np
 import torch
 
 TOP_N = 5
+DEFAULT_MODEL = "amazon-sagemaker-community/xlm-roberta-en-ru-emoji-v2"
 
 def preprocess(text):
     new_text = []
@@ -25,7 +26,11 @@ def get_top_emojis(text, tokenizer, model, top_n=TOP_N):
     return '\t'.join(map(str, emojis))
 
 def main():
-
+    
+    cur_model_name = DEFAULT_MODEL
+    
+    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.
@@ -53,10 +58,12 @@ def main():
         "AlekseyDorkin/xlm-roberta-en-ru-emoji"
     ]
 
-    BASE_MODEL = st.selectbox("Choose a model", models_to_choose)
-
-    tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
-    model = AutoModelForSequenceClassification.from_pretrained(BASE_MODEL)
+    model_name = st.selectbox("Choose a model", models_to_choose)
+    if model_name != cur_model_name:
+        cur_model_name = model_name
+        tokenizer = AutoTokenizer.from_pretrained(cur_model_name)
+        model = AutoModelForSequenceClassification.from_pretrained(cur_model_name)
+        
     
     # Define function to run when submit is clicked
     def submit(message):
@@ -70,9 +77,11 @@ 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__":