improved ux when pipe is loading

app.py

@@ -3,7 +3,7 @@ import transformers
 import matplotlib.pyplot as plt
 
 
-@st.cache(allow_output_mutation=True)
+@st.cache(allow_output_mutation=True, show_spinner=False)
 def get_pipe():
     model_name = "joeddav/distilbert-base-uncased-go-emotions-student"
     model = transformers.AutoModelForSequenceClassification.from_pretrained(model_name)
@@ -21,12 +21,12 @@ st.set_page_config(page_title="Emotion Prediction")
 st.title("Emotion Prediction")
 st.write("Type text into the text box and then press 'Predict' to get the predicted emotion.")
 
-with st.spinner("Loading model..."):
-    pipe = get_pipe()
-
 text = st.text_area('Enter text here:')
 submit = st.button('Predict')
 
+with st.spinner("Loading model..."):
+    pipe = get_pipe()
+
 if (submit and len(text.strip()) > 0) or len(text.strip()) > 0:
 
     prediction = pipe(text)[0]