Spaces:

m-ric
/

beam_search_visualizer

Running on Zero

m-ric HF staff commited on Mar 25

Commit

f93f8e4

•

1 Parent(s): 50a978d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -378,6 +378,7 @@ def get_beam_search_html(input_text, number_steps, number_beams, length_penalty)
         markdown += f"\n- {sequence} (score {outputs.sequences_scores[i]:.2f})"
     print("Markdown output")
     print(markdown)
     original_tree = generate_beams(
         input_text,

         markdown += f"\n- {sequence} (score {outputs.sequences_scores[i]:.2f})"
     print("Markdown output")
     print(markdown)
+    print(input_text, decoded_sequences)
     original_tree = generate_beams(
         input_text,