Update app.py

app.py

@@ -11,6 +11,9 @@ def get_classifier_model():
     return pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
     #return pipeline("zero-shot-classification",model="sentence-transformers/paraphrase-MiniLM-L6-v2")
 
+@st.cache(allow_output_mutation = True)
+def get_summarization_model():
+    return pipeline("summarization", model="facebook/bart-large-mnli", tokenizer="facebook/bart-large-mnli")
 
 #st.sidebar.image("Suncorp-Bank-logo.png",width=255)
 
@@ -20,6 +23,14 @@ st.title("Review Analyzer")
 st.markdown("***")
 
 text = st.text_area(label="Paste/Type the review here..")
+summarize_button_clicked = st.button("Summarize)
+
+if summarize_button_clicked :
+    
+    if text:
+        summarizer = get_summarization_model()
+        st.write(summarizer(text, min_length=5, max_length=40))
+    
 st.markdown("***")
 
 col1, col2, col3 = st.columns((1,1,1))
@@ -54,6 +65,8 @@ def get_classification(candidate_labels):
     )
     return fig
 
+
+
 if classify_button_clicked:
     if text:
         st.markdown("***")