Update app.py

app.py

@@ -1,5 +1,12 @@
 import streamlit as st
 from transformers import pipeline
+from tqdm import tqdm
+import time
+
+def estimate_loading_time():
+    total_steps = 100  # Adjust the total steps as needed
+    for _ in tqdm(range(total_steps), desc="Loading model..."):
+        time.sleep(0.05)  # Simulate loading time
 
 st.set_page_config(page_title='📜✂️ Chat Summarizer 📄🔍')
 st.title('📜✂️ Chat Summarizer 📄🔍')
@@ -29,8 +36,8 @@ if button:
     progress_text = "Model loading. Please wait."
     spinner = st.spinner(progress_text)
     with spinner:
+        estimate_loading_time() 
         pipe = pipeline("summarization", model="nicoladisabato/pegasus-samsum")
-        spinner.stop()
     progress_text = "Summarization in progress. Please wait."
     with st.spinner(progress_text):
         out = pipe(text)