add extra inputs for generation parameters

app.py

@@ -43,6 +43,8 @@ def flatten(l):
 
 
 def run_model_with_frames(text, framesize=4, overlap=3):
+    if overlap > framesize:
+        return "Overlap should be smaller than batch size"
     frames = create_frames(text, framesize, overlap)
     result = set()
     for frame in frames:
@@ -90,9 +92,11 @@ progress and might still be a little bit buggy."""
 with gr.Blocks() as iface:
     gr.Markdown(description)
     context = gr.Textbox(label="Input text")
+    frame_size = gr.Number(value=5, label="Batch size", info="Size of the subparts that are used to generate questions. Increase to speed up the generation", precision=0)
+    overlap = gr.Number(value=4, label="Overlap", info="Overlap between batches. Should be bigger than batch size. Decrease to speed up generation", precision=0)
     questions = gr.Textbox(label="Questions")
     generate_btn = gr.Button("Generate questions")
-    generate_btn.click(fn=run_model_with_frames, inputs=context, outputs=questions, api_name="generate_questions")
+    generate_btn.click(fn=run_model_with_frames, inputs=[context, frame_size, overlap], outputs=questions, api_name="generate_questions")
 
 #iface = gr.Interface(fn=run_model_with_frames, inputs="text", outputs="text")
 iface.launch()