Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -174,6 +174,9 @@ STYLE = """
|
|
174 |
.nonselected-sequence {
|
175 |
background-color: var(--primary-500);
|
176 |
}
|
|
|
|
|
|
|
177 |
"""
|
178 |
|
179 |
|
@@ -216,10 +219,10 @@ def generate_nodes(node, step):
|
|
216 |
selected_class = "selected-sequence"
|
217 |
else:
|
218 |
selected_class = "nonselected-sequence"
|
219 |
-
return f"<li> <a
|
220 |
|
221 |
html_content = (
|
222 |
-
f"<li> <a
|
223 |
)
|
224 |
if node.table is not None:
|
225 |
html_content += node.table
|
@@ -238,7 +241,7 @@ def generate_nodes(node, step):
|
|
238 |
def generate_html(start_sentence, original_tree):
|
239 |
html_output = f"""<div class="custom-container">
|
240 |
<div class="tree">
|
241 |
-
<ul> <li> <a
|
242 |
html_output += "<ul> "
|
243 |
for subnode in original_tree.children.values():
|
244 |
html_output += generate_nodes(subnode, step=1)
|
@@ -487,7 +490,7 @@ This parameter will not impact the beam search paths, but only influence the cho
|
|
487 |
)
|
488 |
with gr.Row():
|
489 |
n_steps = gr.Slider(
|
490 |
-
label="Number of steps", minimum=1, maximum=
|
491 |
)
|
492 |
n_beams = gr.Slider(
|
493 |
label="Number of beams", minimum=2, maximum=4, step=1, value=4
|
|
|
174 |
.nonselected-sequence {
|
175 |
background-color: var(--primary-500);
|
176 |
}
|
177 |
+
.nopadding {
|
178 |
+
padding-left: 0;
|
179 |
+
}
|
180 |
"""
|
181 |
|
182 |
|
|
|
219 |
selected_class = "selected-sequence"
|
220 |
else:
|
221 |
selected_class = "nonselected-sequence"
|
222 |
+
return f"<li> <a class='end-of-text child {selected_class}'> <span> <b>{clean(token)}</b> <br>Total score: {node.total_score:.2f}</span> </a> </li>"
|
223 |
|
224 |
html_content = (
|
225 |
+
f"<li> <a class='nonfinal child'> <span> <b>{clean(token)}</b> </span>"
|
226 |
)
|
227 |
if node.table is not None:
|
228 |
html_content += node.table
|
|
|
241 |
def generate_html(start_sentence, original_tree):
|
242 |
html_output = f"""<div class="custom-container">
|
243 |
<div class="tree">
|
244 |
+
<ul> <li> <a id='root' class="nopadding"> <span> <b>{start_sentence}</b> </span> {original_tree.table} </a>"""
|
245 |
html_output += "<ul> "
|
246 |
for subnode in original_tree.children.values():
|
247 |
html_output += generate_nodes(subnode, step=1)
|
|
|
490 |
)
|
491 |
with gr.Row():
|
492 |
n_steps = gr.Slider(
|
493 |
+
label="Number of steps", minimum=1, maximum=12, step=1, value=5
|
494 |
)
|
495 |
n_beams = gr.Slider(
|
496 |
label="Number of beams", minimum=2, maximum=4, step=1, value=4
|