Update app.py
Browse files
app.py
CHANGED
@@ -150,7 +150,7 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
|
|
150 |
btn.click(partial(run_interpretation, i=i), [global_state, interpretation_prompt, num_tokens, do_sample, temperature,
|
151 |
top_k, top_p, repetition_penalty, length_penalty
|
152 |
], [json_output])
|
153 |
-
|
154 |
json_output.render()
|
155 |
|
156 |
original_prompt_btn.click(get_hidden_states, [original_prompt_raw], [global_state, *tokens_container])
|
|
|
150 |
btn.click(partial(run_interpretation, i=i), [global_state, interpretation_prompt, num_tokens, do_sample, temperature,
|
151 |
top_k, top_p, repetition_penalty, length_penalty
|
152 |
], [json_output])
|
153 |
+
tokens_container.append(btn)
|
154 |
json_output.render()
|
155 |
|
156 |
original_prompt_btn.click(get_hidden_states, [original_prompt_raw], [global_state, *tokens_container])
|