Update app.py

app.py

@@ -27,8 +27,14 @@ def get_sentiment(text):
     dec = [sentiment_tokenizer.decode(ids) for ids in output]
     label = dec[0]
     return label
+
+@st.cache    
+def get_summarizer():
+    summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+    return summarizer
   
 sentiment_model   = get_sentiment_model()
+summarizer = get_summarizer()
 text_generator = load_text_gen_model()
 
 action = st.sidebar.selectbox("Pick an Action", ["Analyse a Review","Generate an Article","Create an Image"])
@@ -40,4 +46,10 @@ if action == "Analyse a Review":
         #st.write(res)
         sentiment = get_sentiment(review)
         st.write(sentiment)
+        
+        if st.button("Summarize the review"):
+            summary = summarizer(ARTICLE, max_length=130, min_length=30, do_sample=False)
+            st.write(summary)
+
+