updates

src/visualization/visualize.py

@@ -0,0 +1,22 @@
+import streamlit as st
+from src.models.predict_model import predict_model
+
+
+def visualize():
+    st.write('# Summarization  UI')
+    st.markdown(
+        '''
+        *For additional questions and inquiries, please contact **Gagan Bhatia** via [LinkedIn](
+        https://www.linkedin.com/in/gbhatia30/) or [Github](https://github.com/gagan3012).*
+        ''')
+
+    text = st.text_area("Enter text here")
+    if st.button("Generate Summary"):
+        with st.spinner("Connecting the Dots..."):
+            text = predict_model(text=text)
+        st.write("# Generated Summary:")
+        st.write("{}".format(text))
+
+
+if __name__ == "__main__":
+    visualize()