Remove length penalty, num return sequence fields

app.py

@@ -273,7 +273,7 @@ class BeamNode:
     is_selected_sequence: bool
 
 
-def generate_beams(n_beams, start_sentence, scores, length_penalty, decoded_sequences):
+def generate_beams(n_beams, start_sentence, scores, length_penalty=1, decoded_sequences=1):
     original_tree = BeamNode(
         cumulative_score=0,
         current_token_ix=None,
@@ -421,7 +421,7 @@ def generate_beams(n_beams, start_sentence, scores, length_penalty, decoded_sequ
 
 @spaces.GPU
 def get_beam_search_html(
-    input_text, number_steps, number_beams, length_penalty, num_return_sequences
+    input_text, number_steps, number_beams
 ):
     inputs = tokenizer([input_text], return_tensors="pt")
 
@@ -429,9 +429,7 @@ def get_beam_search_html(
         **inputs,
         max_new_tokens=number_steps,
         num_beams=number_beams,
-        num_return_sequences=min(num_return_sequences, number_beams),
         return_dict_in_generate=True,
-        length_penalty=length_penalty,
         output_scores=True,
         do_sample=False,
     )
@@ -443,16 +441,12 @@ def get_beam_search_html(
             number_beams,
             input_text,
             outputs.scores[:],
-            length_penalty,
-            decoded_sequences,
         )
     else:
         original_tree = generate_beams(
             number_beams,
             input_text,
             outputs.scores,
-            0,
-            decoded_sequences,
         )
         
     html = generate_html(input_text, original_tree)
@@ -499,22 +493,16 @@ This parameter will not impact the beam search paths, but only influence the cho
         n_beams = gr.Slider(
             label="Number of beams", minimum=1, maximum=4, step=1, value=4
         )
-        length_penalty = gr.Slider(
-            label="Length penalty", minimum=-3, maximum=3, step=0.5, value=1
-        )
-        num_return_sequences = gr.Slider(
-            label="Number of return sequences", minimum=1, maximum=4, step=1, value=3
-        )
 
     n_beams.change(
-        fn=change_num_return_sequences, inputs=n_beams, outputs=num_return_sequences
+        fn=change_num_return_sequences, inputs=n_beams, outputs=1
     )
     button = gr.Button()
     out_html = gr.Markdown()
     out_markdown = gr.Markdown()
     button.click(
         get_beam_search_html,
-        inputs=[text, n_steps, n_beams, length_penalty, num_return_sequences],
+        inputs=[text, n_steps, n_beams],
         outputs=[out_html, out_markdown],
     )