Update app.py

app.py

@@ -3,10 +3,6 @@ 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 📄🔍')
@@ -14,7 +10,7 @@ st.title('📜✂️ Chat Summarizer 📄🔍')
 
 # Define available models and their corresponding names
 models = {
-    "PEGASUS fine-tuned": "nicoladisabato/pegasus-samsum"
+    "nicoladisabato/pegasus-samsum": "nicoladisabato/pegasus-samsum"
 }
 
 selected_model = st.selectbox("Select a model", list(models.keys()))
@@ -27,18 +23,16 @@ Nico: So won't you come?
 Demo: I prefer to go to Lido Le Canne in Pulsano.
 Nico: Ahahah
 """
-progress_text = "Model loading. Please wait."
 
 text = st.text_area("Enter your chat", value=predefined_text, height=200)
 button = st.button('Summarize')
 
 if button:
-    progress_text = "Model loading. Please wait."
+    progress_text = "Loading the pipeline. It will take a minute :)"
     spinner = st.spinner(progress_text)
     with spinner:
-        estimate_loading_time() 
         pipe = pipeline("summarization", model="nicoladisabato/pegasus-samsum")
-    progress_text = "Summarization in progress. Please wait."
+    progress_text = "Summarization in progress."
     with st.spinner(progress_text):
         out = pipe(text)
         st.json(out)