Update app.py

app.py

@@ -13,10 +13,10 @@ def correct_text(text, genConfig):
     corrected_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return corrected_text
 
-def respond(text, max_length, min_length, max_new_tokens, min_new_tokens, num_beams, num_beam_groups, temperature, top_k, top_p):
+def respond(text, max_new_tokens, min_new_tokens, num_beams, num_beam_groups, temperature, top_k, top_p):
     config = GenerationConfig(
-        max_length=max_length,
-        min_length=min_length,
+        max_new_tokens=max_new_tokens,
+        min_new_tokens=min_new_tokens,
         num_beams=num_beams,
         num_beam_groups=num_beam_groups,
         temperature=float(temperature),
@@ -25,12 +25,6 @@ def respond(text, max_length, min_length, max_new_tokens, min_new_tokens, num_be
         early_stopping=True,
         do_sample=True
     )
-
-    # Add max/min new tokens if they are there
-    if max_new_tokens > 0: 
-        config.max_new_tokens = max_new_tokens
-    if min_new_tokens > 0: 
-        config.min_new_tokens = min_new_tokens
     
     corrected = correct_text(text, config)
     yield corrected
@@ -56,8 +50,6 @@ with gr.Blocks() as demo:
     submitBtn = gr.Button("Submit")
     
     with gr.Accordion("Generation Parameters:", open=False):
-        max_length  = gr.Slider(minimum=1,   maximum=256,   value=80,  step=1,    label="Max Length")
-        min_length  = gr.Slider(minimum=1,   maximum=256,   value=0,   step=1,    label="Min Length")
         max_tokens  = gr.Slider(minimum=0,   maximum=256,   value=0,   step=1,    label="Max New Tokens")
         min_tokens  = gr.Slider(minimum=0,   maximum=256,   value=0,   step=1,    label="Min New Tokens")
         num_beams   = gr.Slider(minimum=1,   maximum=20,    value=5,   step=1,    label="Num Beams")
@@ -68,6 +60,6 @@ with gr.Blocks() as demo:
         
 
     
-    submitBtn.click(respond, [prompt_box, max_length, min_length, max_tokens, min_tokens, num_beams, beam_groups, temperature, top_k, top_p], output_box)
+    submitBtn.click(respond, [prompt_box, max_tokens, min_tokens, num_beams, beam_groups, temperature, top_k, top_p], output_box)
 
 demo.launch()