Spaces:

m-ric
/

beam_search_visualizer

Running on Zero

m-ric HF staff commited on Apr 2

Commit

939822c

•

1 Parent(s): c4a1c93

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -456,7 +456,7 @@ def get_beam_search_html(
     else:
         original_tree = generate_beams(
             n_beams,
-            start_sentence,
             outputs.logits,
             0,
             decoded_sequences,

     else:
         original_tree = generate_beams(
             n_beams,
+            input_text,
             outputs.logits,
             0,
             decoded_sequences,