add temperature and count slider

app.py

@@ -11,6 +11,23 @@ st.title('Question Generator by Eddevs')
 
 select = st.selectbox('Type',  ['Question Generator', 'Paraphrasing', 'Summarization', 'Fill in the blank'])
 
+
+if select == "Question Generator":
+    with st.form("question_gen"):
+        # left_column, right_column = st.columns(2)
+        # left_column.selectbox('Type',  ['Question Generator', 'Paraphrasing'])
+        #st.selectbox('Model',  ['T5', 'GPT Neo-X'])
+
+        text_input = st.text_area("Input Text")
+
+        submitted = st.form_submit_button("Generate")
+
+        if submitted:
+            with st.spinner('Wait for it...'):
+                result = FillInSummary().summarize(text_input)
+            st.write(text_input)
+
+
 if select == "Summarization":
     with st.form("summarization"):
         # left_column, right_column = st.columns(2)
@@ -28,7 +45,7 @@ if select == "Summarization":
 
 
 if select == "Fill in the blank":
-    with st.form("summarization"):
+    with st.form("fill_in_the_blank"):
         text_input = st.text_area("Input Text")
 
         submitted = st.form_submit_button("Generate")
@@ -44,14 +61,16 @@ if select == "Fill in the blank":
 if select == "Paraphrasing":
     with st.form("paraphrasing"):
         # st.selectbox('Model',  ['T5', 'GPT Neo-X'])
-
+        left_column, right_column = st.columns(2)
+        count = left_column.slider('Count', 0, 10, 3)
+        temperature = right_column.slider('Temperature', 0.0, 10.0, 1.5)
         text_input = st.text_area("Input Text")
 
         submitted = st.form_submit_button("Generate")
 
         if submitted:
             with st.spinner('Wait for it...'):
-                paraphrase_model = PegasusParaphraser()
+                paraphrase_model = PegasusParaphraser(num_return_sequences=count,temperature=temperature)
                 result = paraphrase_model.paraphrase(text_input)
             st.write(result)