Update app.py

app.py

@@ -76,6 +76,15 @@ def main():
                     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]) 
+                    
+             elif model_type == "Bart + T5":
+                text_to_summarize = clean_text
+
+                with st.spinner(
+                    text="Loading Bart + T5 Models and Extracting summary. This might take a few seconds depending on the length of your text..."):
+                    summarizer_model = bart_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: