decoding_visualizer

Running on T4

m-ric commited on Apr 2, 2024

Commit

939822c

verified ·

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,