Update app.py

app.py

@@ -156,7 +156,7 @@ def generate_markdown_table(scores, top_k=4, chosen_tokens=None):
     return markdown_table
 
 
-def display_tree(scores, sequences, beam_indices):
+def display_tree(start_sentence, scores, sequences, beam_indices):
     display = """<div class="container">
 				<div class="tree">
                 <ul>"""
@@ -184,7 +184,7 @@ def display_tree(scores, sequences, beam_indices):
 
             current_tree = current_tree["children"][current_token_choice]
 
-    display += generate_html("Today is", original_tree)
+    display += generate_html(start_sentence, original_tree)
 
     display += """
         </ul>
@@ -210,6 +210,7 @@ def get_tables(input_text, number_steps, number_beams):
     )
 
     tables = display_tree(
+        input_text,
         outputs.scores,
         outputs.sequences[:, len(inputs) :],
         outputs.beam_indices[:, : -len(inputs)],