Update app.py

app.py

@@ -47,39 +47,43 @@ example_prompts = [
     "I envy people that know love.",
     "Nature is so beautiful"]
 
-example = st.selectbox("Choose an example", example_prompts)
-
-# Take the message which needs to be processed
-message = st.text_area("...or paste some text to see the model's predictions", example)
-# st.title(message)
-st.text('')
-models_to_choose = [
-    "amazon-sagemaker-community/xlm-roberta-en-ru-emoji-v2",
-    "AlekseyDorkin/xlm-roberta-en-ru-emoji"
-]
-
-model_name = st.selectbox("Choose a model", models_to_choose)
-if model_name != cur_model_name:
-    print("reloading model")
-    cur_model_name = model_name
-    tokenizer = AutoTokenizer.from_pretrained(cur_model_name)
-    model = AutoModelForSequenceClassification.from_pretrained(cur_model_name)
+def main():
+    example = st.selectbox("Choose an example", example_prompts)
     
-
-# 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))
-    else:
-        st.error("The text can't be empty")
-
-# Run algo when submit button is clicked
-if st.button('Submit'):
-    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,
-)
+    # Take the message which needs to be processed
+    message = st.text_area("...or paste some text to see the model's predictions", example)
+    # st.title(message)
+    st.text('')
+    models_to_choose = [
+        "amazon-sagemaker-community/xlm-roberta-en-ru-emoji-v2",
+        "AlekseyDorkin/xlm-roberta-en-ru-emoji"
+    ]
+    
+    model_name = st.selectbox("Choose a model", models_to_choose)
+    if model_name != cur_model_name:
+        print("reloading model")
+        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):
+        if len(message) > 0:
+            st.header(get_top_emojis(message, tokenizer=tokenizer, model=model))
+        else:
+            st.error("The text can't be empty")
+    
+    # Run algo when submit button is clicked
+    if st.button('Submit'):
+        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,
+    )
+    
+if __name__ == "__main__":
+    main()