Update app.py

app.py

@@ -2,15 +2,6 @@ import streamlit as st
 from huggingface_hub import hf_hub_download
 from utils import *
 
-# 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'])
-
 # APP
 st.set_page_config(
     page_title="Hengam",
@@ -40,6 +31,25 @@ 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)
+
+st.empty()  # Clear the "Please wait" message
+
 st.markdown("")
 st.markdown("## **📌 Paste any Persian (Farsi) text you want to extract its temporal markers.**")
 with st.form(key="my_form"):
@@ -75,5 +85,5 @@ with st.form(key="my_form"):
 if submit_button:
     result = ner(doc)
     st.write("")  # Add vertical spacing
-    st.markdown("## **🎈Check results**")
+    st.markdown("##Result**")
     st.code(result, language="python")