beam_search_visualizer

Runtime error

m-ric HF staff commited on Mar 21

Commit

2eb8692

•

1 Parent(s): a7a4e14

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,6 @@
 STYLE = """
-body {
-	height: 100vh;
-	width: 100vw;
 	display: grid;
 	align-items: center;
 }
@@ -147,9 +146,7 @@ def generate_markdown_table(scores, top_k=4, chosen_tokens=None):
 def display_tree(scores, sequences, beam_indices):
-    display = """<body>
-		<div class="container">
-			<div class="row">
 				<div class="tree">
                 <ul>"""
     sequences = sequences.cpu().numpy()
@@ -181,8 +178,6 @@ def display_tree(scores, sequences, beam_indices):
     display += """
         </ul>
         </div>
-    </div>
-    </div>
     </body>
     """
     print(display)

 STYLE = """
+.container {
+	width: 100%;
 	display: grid;
 	align-items: center;
 }
 def display_tree(scores, sequences, beam_indices):
+    display = """<div class="container">
 				<div class="tree">
                 <ul>"""
     sequences = sequences.cpu().numpy()
     display += """
         </ul>
         </div>
     </body>
     """
     print(display)