sidebar loads before model

app.py

@@ -17,19 +17,6 @@ urls = {
 }
 
 st.header("Extracting a Knowledge Base from text")
-st_model_load = st.text('Loading NER model... It may take a while.')
-
-@st.cache(allow_output_mutation=True)
-def load_model():
-    print("Loading model...")
-    tokenizer = AutoTokenizer.from_pretrained("Babelscape/rebel-large")
-    model = AutoModelForSeq2SeqLM.from_pretrained("Babelscape/rebel-large")
-    print("Model loaded!")
-    return tokenizer, model
-
-tokenizer, model = load_model()
-st.success('Model loaded!')
-st_model_load.text("")
 
 # sidebar
 with st.sidebar:
@@ -46,6 +33,21 @@ with st.sidebar:
     st.header("Considerations")
     st.markdown("If you look closely at the extracted knowledge graphs, some extracted relations are false. Indeed, relation extraction models are still far from perfect and require further steps in the pipeline to build reliable knowledge graphs. Consider this demo as a starting step!")
 
+# Loading the model
+st_model_load = st.text('Loading NER model... It may take a while.')
+
+@st.cache(allow_output_mutation=True)
+def load_model():
+    print("Loading model...")
+    tokenizer = AutoTokenizer.from_pretrained("Babelscape/rebel-large")
+    model = AutoModelForSeq2SeqLM.from_pretrained("Babelscape/rebel-large")
+    print("Model loaded!")
+    return tokenizer, model
+
+tokenizer, model = load_model()
+st.success('Model loaded!')
+st_model_load.text("")
+
 # Choose from where to generate the KB
 options = [
     "Text",