fill in the gap, summarization

app.py

@@ -1,32 +1,67 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import pipeline, PegasusForConditionalGeneration, PegasusTokenizer
+from fill_in_summary import FillInSummary
+
+def paraphrase(text):
+    return text
+
 
 st.title('Question Generator by Eddevs')
 
-left_column, right_column = st.columns(2)
-left_column.selectbox('Type',  ['Question Generator', 'Paraphrasing'])
-right_column.selectbox('Question Generator',  ['T5', 'GPT Neo-X'])
-
-input = st.text_area("Input Text")
-
-def summarize(text):
-  # Refer to https://huggingface.co/docs/transformers/v4.18.0/en/main_classes/pipelines#transformers.SummarizationPipeline
-  # for further information about configuration.
-  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
-  # Refer to https://huggingface.co/docs/transformers/main/en/main_classes/configuration#transformers.PretrainedConfig
-  # for further configuration of of the
-  output: list = summarizer(
-      text,
-      max_length=130,
-      min_length=30,
-      do_sample=False)
-  return output
-  #return output[0]['summary_text']
-
-if st.button('Generate'):
+select = st.selectbox('Type',  ['Question Generator', 'Paraphrasing', 'Summarization', 'Fill in the gap'])
+
+if select == "Summarization":
+    with st.form("summarization"):
+        # left_column, right_column = st.columns(2)
+        # left_column.selectbox('Type',  ['Question Generator', 'Paraphrasing'])
+        st.selectbox('Model',  ['T5', 'GPT Neo-X'])
+
+        input = st.text_area("Input Text")
+
+        submitted = st.form_submit_button("Generate")
+
+        if submitted:
+            st.write(FillInSummary().summarize(input))
+
+
+if select == "Fill in the gap":
+    with st.form("summarization"):
+        # left_column, right_column = st.columns(2)
+        # left_column.selectbox('Type',  ['Question Generator', 'Paraphrasing'])
+        st.selectbox('Model',  ['T5', 'GPT Neo-X'])
+
+        input = st.text_area("Input Text")
+
+        submitted = st.form_submit_button("Generate")
+
+        if submitted:
+            fill = FillInSummary()
+            summarized = fill.summarize(input)
+            st.write(fill.blank_ne_out(summarized))
+
+
+if select == "Paraphrasing":
+    with st.form("paraphrasing"):
+        # left_column, right_column = st.columns(2)
+        # left_column.selectbox('Type',  ['Question Generator', 'Paraphrasing'])
+        st.selectbox('Model',  ['T5', 'GPT Neo-X'])
+
+        input = st.text_area("Input Text")
+
+        submitted = st.form_submit_button("Generate")
+
+        if submitted:
+            st.write(paraphrase(input))
+
+
+
+
+
+#if st.button('Generate'):
     # st.write(input)
-    st.write(summarize(input))
-    st.success("We have generated 105 Questions for you")
+    #st.success("We have generated 105 Questions for you")
     # st.snow()
 ##else:
     ##nothing here
+
+

requirements.txt

@@ -1,3 +1,5 @@
-transformers
+transformers~=4.18.0
 torch
-tensorflow
+tensorflow
+streamlit~=1.8.1
+sentencepiece==0.1.96