fix missing import, some cleanup

app.py

@@ -8,13 +8,13 @@ def paraphrase(text):
 
 st.title('Question Generator by Eddevs')
 
-select = st.selectbox('Type',  ['Question Generator', 'Paraphrasing', 'Summarization', 'Fill in the gap'])
+select = st.selectbox('Type',  ['Question Generator', 'Paraphrasing', 'Summarization', 'Fill in the blank'])
 
 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'])
+        #st.selectbox('Model',  ['T5', 'GPT Neo-X'])
 
         input = st.text_area("Input Text")
 
@@ -24,12 +24,8 @@ if select == "Summarization":
             st.write(FillInSummary().summarize(input))
 
 
-if select == "Fill in the gap":
+if select == "Fill in the blank":
     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")
@@ -42,8 +38,6 @@ if select == "Fill in the gap":
 
 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")

fill_in_summary.py

@@ -1,3 +1,4 @@
+from transformers import pipeline
 
 class FillInSummary:
     """Organizing Summarization and Subsequent Fill-In-Task."""