Update app.py

app.py

@@ -8,8 +8,7 @@ sentiment_classifier = pipeline("text-classification",model='Ryleeeee/CustomSent
 summarizer = pipeline("summarization", model="MurkatG/review-summarizer-en")
 
 # Streamlit application title and background image
-# st.image("./header.png", width=300, use_column_width=False)
-st.markdown("<div style='text-align:center'><img src='./header.png' width='300'></div>", unsafe_allow_html=True)
+st.image("./header.png", use_column_width=False)
 st.markdown("<h1 style='text-align: center;'>Customer Review Analysis</h1>", unsafe_allow_html=True)
 
 st.write("Setiment classification: positive, netural, negative")
@@ -43,5 +42,9 @@ if st.button("Classify Sentiment"):
         st.write("This review sentiment is ", sentiment_result[1])
         st.write("Prediction score is ", sentiment_result[0])
 
-if sentiment_result[1] == 'negative':
-    st.button("Summarize Review")
+        if sentiment_result[1] == 'negative':
+            st.button("Summarize Review")
+            if st.button("Summarize Review"):
+                summary = summarize_text(review)
+                st.write("Review summary: ", summary)
+