Update app.py

app.py

@@ -31,24 +31,22 @@ with st.expander("ℹ️ - About this app", expanded=True):
     )
     st.markdown("")
 
- # Display initial loading message
-st.write("Please wait while the model is loading...")
-
-# Download the models
-HengamTransW = hf_hub_download(repo_id="kargaranamir/Hengam", filename="HengamTransW.pth")
-HengamTransA = hf_hub_download(repo_id="kargaranamir/Hengam", filename="HengamTransA.pth")
-
-
+    
 # Load models lazily
 @st.cache(allow_output_mutation=True)
 def load_ner_model(model_path):
     return NER(model_path=model_path, tags=['B-TIM', 'I-TIM', 'B-DAT', 'I-DAT', 'O'])
 
-# cache
-load_ner_model(HengamTransW)
-load_ner_model(HengamTransA)
+with st.spinner(text="Please wait while the model is loading...."):
+
+    # Download the models
+    HengamTransW = hf_hub_download(repo_id="kargaranamir/Hengam", filename="HengamTransW.pth")
+    HengamTransA = hf_hub_download(repo_id="kargaranamir/Hengam", filename="HengamTransA.pth")
+
+    # cache
+    load_ner_model(HengamTransW)
+    load_ner_model(HengamTransA)
 
-st.empty()  # Clear the "Please wait" message
 
 st.markdown("")
 st.markdown("## **📌 Paste any Persian (Farsi) text you want to extract its temporal markers.**")
@@ -56,7 +54,6 @@ with st.form(key="my_form"):
     c1, c2 = st.columns([2, 3])
     
     with c1:
-        st.write("Please wait while loading the model...")
         model_paths = {
             "HengamTransW.pth": HengamTransW,
             "HengamTransA.pth": HengamTransA,
@@ -71,7 +68,6 @@ with st.form(key="my_form"):
         )
         
         ner = load_ner_model(model_paths[ModelType])
-        st.empty()  # Clear the "Please wait" message
 
     with c2:
         doc = st.text_area(
@@ -83,7 +79,7 @@ with st.form(key="my_form"):
         submit_button = st.form_submit_button(label="✨ Extract Temporal Markers!")
 
 if submit_button:
-    result = ner(doc)
+    result = ner("# " + doc)
     st.write("")  # Add vertical spacing
     st.markdown("##Result**")
     st.code(result, language="python")