Initial app setup

.gitignore

@@ -0,0 +1,2 @@
+.idea/
+venv/

app.py

@@ -0,0 +1,50 @@
+from textwrap import wrap 
+
+from transformers import pipeline
+import streamlit as st
+
+st.markdown('# Terms and conditions abstractive summarization model :pencil:')
+st.write('This app summarizes the provided terms and conditions. '
+         'Terms and conditions Summarization model is based on sshleifer/distilbart-cnn-6-6')
+
+st.markdown("""
+To use this:
+- Copy terms and conditions and hit 'Summarize':point_down:""")
+
+
+@st.cache(allow_output_mutation=True,
+          suppress_st_warning=True,
+          show_spinner=False)
+def load_model():
+    with st.spinner('Please wait for the model to load...'):
+        terms_and_conditions_pipeline = pipeline(
+            'terms-and-conditions-summarizer',
+            model='ml6team/distilbart-tos-summarizer-tosdr',
+            tokenizer='ml6team/distilbart-tos-summarizer-tosdr'
+        )
+    return terms_and_conditions_pipeline
+
+
+tc_pipeline = load_model()
+
+if 'text' not in st.session_state:
+    st.session_state['text'] = ""
+
+left_area, right_area = st.columns(2)
+
+left_area.header("Input")
+form = left_area.form(key='terms-and-conditions')
+placeholder = form.empty()
+placeholder.empty()
+input_text = placeholder.text_area(value=st.session_state.text, label='Insert text:', key='input_text')
+submit_button = form.form_submit_button(label='Summarize')
+
+# Right area
+right_area.header("Output")
+
+if submit_button:
+    base_text = st.session_state.input_text
+    output_text = " ".join([x['generated_text'] for x in tc_pipeline(wrap(base_text, 128))])
+    right_area.markdown('#####')
+    right_area.text_area(value=output_text, label="Summary:")
+

requirements.txt

@@ -0,0 +1,5 @@
+nlpaug==1.1.7
+streamlit==1.0.0
+torch==1.9.1
+torchvision==0.10.1
+transformers==4.10.3