Update app.py

app.py

@@ -39,7 +39,7 @@ def main():
         st.sidebar.markdown("<h4 style = 'color:green; align:center; font-size: 20px;'> Model Settings</h1>", unsafe_allow_html=True)
         max_length= st.sidebar.slider("Maximum length of the generated text",min_value=100,max_value=500)
         min_length= st.sidebar.slider("Minimum length of the generated text",min_value=30)
-        model_type = st.sidebar.selectbox("Model type", options=["Bart","T5",])
+        model_type = st.sidebar.selectbox("Model type", options=["Bart","T5" "ssr",])
         
         # This function is used to upload a .txt, .pdf, .docx file for summarization
         upload_doc = st.file_uploader("Upload a .txt, .pdf, .docx file for summarisation")
@@ -77,7 +77,16 @@ def main():
                     summarized_text = summarizer_model(text_to_summarize, max_length=max_length, min_length=min_length)
                     summarized_text = ' '.join([summ['summary_text'] for summ in summarized_text])                      
 
-            res_col1 ,res_col2 = st.columns(2)
+            elif model_type == "ssr":
+                text_to_summarize = clean_text
+
+                with st.spinner(
+                    text="Loading SSR Model and Extracting summary. This might take a few seconds depending on the length of your text..."):
+                    summarizer_model = t5()
+                    summarized_text = summarizer_model(text_to_summarize, max_length=max_length, min_length=min_length)
+                    summarized_text = ' '.join([summ['summary_text'] for summ in summarized_text])
+          
+              res_col1 ,res_col2 = st.columns(2)
             with res_col1:
                 st.subheader("Generated Text Visualisation")
                 # Create and generate a word cloud image: