Update app.py

app.py

@@ -5,7 +5,28 @@ from parstdex import Parstdex
 # set config
 st.set_page_config(layout="wide", page_title="Parstdex")
 
-model = Parstdex()
+# Layout adjustments
+st.markdown(
+    """
+<style>
+.reportview-container .main .block-container {
+    max-width: 1400px;
+}
+</style>
+""",
+    unsafe_allow_html=True,
+)
+
+
+# Load models lazily
+@st.cache(allow_output_mutation=True)
+def load_model():
+    model = Parstdex()
+    return model
+
+with st.spinner(text="Please wait while the model is loading...."):
+    # cache
+    model = Parstdex()
 
 # APP
 # set title and subtitle
@@ -27,10 +48,10 @@ with st.spinner("Processing Text  and Extracting..."):
 
         # display the result
         st.markdown("**Span:**")
-        st.write(spans)
+        st.code(spans, language="python")
         st.markdown("**Marker:**")
-        st.write(markers)
+        st.code(markers, language="python")
         st.markdown("**Time_ML:**")
-        st.write(f"\u202b{time_mls}\u202c")
+        st.code(f"\u202b{time_mls}\u202c", language="python")
         st.markdown("**NER:**")
-        st.write(ners)
+        st.code(ners, language="python")