beam_search_visualizer

Running

jvamvas commited on Apr 8

Commit

10f448e

•

1 Parent(s): f676751

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -441,7 +441,7 @@ def get_beam_search_html(
             number_beams,
             input_text,
             outputs.scores[:],
-            length_penalty=1,
             decoded_sequences,
         )
     else:
@@ -449,7 +449,7 @@ def get_beam_search_html(
             number_beams,
             input_text,
             outputs.scores,
-            length_penalty=0,
             decoded_sequences,
         )

             number_beams,
             input_text,
             outputs.scores[:],
+            1,
             decoded_sequences,
         )
     else:
             number_beams,
             input_text,
             outputs.scores,
+            0,
             decoded_sequences,
         )