Update app.py

app.py

@@ -273,7 +273,7 @@ class BeamNode:
     is_selected_sequence: bool
 
 
-def generate_beams(n_beams, start_sentence, scores, length_penalty=1, decoded_sequences=1):
+def generate_beams(n_beams, start_sentence, scores, length_penalty, decoded_sequences):
     original_tree = BeamNode(
         cumulative_score=0,
         current_token_ix=None,
@@ -441,12 +441,16 @@ def get_beam_search_html(
             number_beams,
             input_text,
             outputs.scores[:],
+            length_penalty=1,
+            decoded_sequences,
         )
     else:
         original_tree = generate_beams(
             number_beams,
             input_text,
             outputs.scores,
+            length_penalty=0,
+            decoded_sequences,
         )
         
     html = generate_html(input_text, original_tree)