Update app.py

app.py

@@ -1,8 +1,16 @@
 import streamlit as st
 from transformers import pipeline
 
-st.set_page_config(page_title='🖊📄 Chat Summarization WebApp')
-st.title('🖊📄 Chat Summarization WebApp')
+st.set_page_config(page_title='📜✂️ Chat Summarizer 📄🔍')
+st.title('📜✂️ Chat Summarizer 📄🔍')
+
+
+# Define available models and their corresponding names
+models = {
+    "PEGASUS fine-tuned": "nicoladisabato/pegasus-samsum"
+}
+
+selected_model = st.selectbox("Select a model", list(models.keys()))
 
 predefined_text = """Nico: the moon is very beautiful tonight.
 Demo: I agree with you.
@@ -19,9 +27,11 @@ button = st.button('Summarize')
 
 if button:
     progress_text = "Model loading. Please wait."
-    with st.spinner(progress_text):
+    spinner = st.spinner(progress_text)
+    with spinner:
         pipe = pipeline("summarization", model="nicoladisabato/pegasus-samsum")
-        progress_text = "Summarization in progress. Please wait."
-        with st.spinner(progress_text):
-            out = pipe(text)
-            st.json(out)
+        spinner.stop()
+    progress_text = "Summarization in progress. Please wait."
+    with st.spinner(progress_text):
+        out = pipe(text)
+        st.json(out)