Update app.py
Browse files
app.py
CHANGED
|
@@ -47,9 +47,13 @@ model_info = {
|
|
| 47 |
}
|
| 48 |
|
| 49 |
|
| 50 |
-
suggested_interpretation_prompts = [
|
|
|
|
|
|
|
|
|
|
| 51 |
"Before responding, let me repeat the message you wrote:",
|
| 52 |
-
"Let me repeat the message:"
|
|
|
|
| 53 |
|
| 54 |
|
| 55 |
## functions
|
|
@@ -194,13 +198,7 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
|
|
| 194 |
tokens_container.append(btn)
|
| 195 |
progress_dummy = gr.Markdown('', elem_id='progress_dummy')
|
| 196 |
|
| 197 |
-
|
| 198 |
-
interpretation_bubbles = [gr.Textbox('', container=False, visible=False, elem_classes=['bubble',
|
| 199 |
-
'even_bubble' if i % 2 == 0 else 'odd_bubble'])
|
| 200 |
-
for i in range(model.config.num_hidden_layers)]
|
| 201 |
-
|
| 202 |
-
|
| 203 |
-
with gr.Accordion(open=False, label='Settings'):
|
| 204 |
with gr.Row():
|
| 205 |
num_tokens = gr.Slider(1, 100, step=1, value=20, label='Max. # of Tokens')
|
| 206 |
repetition_penalty = gr.Slider(1., 10., value=1, label='Repetition Penalty')
|
|
@@ -213,6 +211,12 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
|
|
| 213 |
top_k = gr.Slider(1, 1000, value=50, step=1, label='top k')
|
| 214 |
top_p = gr.Slider(0., 1., value=0.95, label='top p')
|
| 215 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 216 |
# with gr.Group():
|
| 217 |
# with gr.Row():
|
| 218 |
# for txt in model_info.keys():
|
|
|
|
| 47 |
}
|
| 48 |
|
| 49 |
|
| 50 |
+
suggested_interpretation_prompts = [
|
| 51 |
+
"Sure, here's a bullet list of the key words in your message:",
|
| 52 |
+
"Sure, I'll summarize your message:",
|
| 53 |
+
"Sure, here are the words in your message:",
|
| 54 |
"Before responding, let me repeat the message you wrote:",
|
| 55 |
+
"Let me repeat the message:"
|
| 56 |
+
]
|
| 57 |
|
| 58 |
|
| 59 |
## functions
|
|
|
|
| 198 |
tokens_container.append(btn)
|
| 199 |
progress_dummy = gr.Markdown('', elem_id='progress_dummy')
|
| 200 |
|
| 201 |
+
with gr.Accordion(open=False, label='Generation Settings'):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 202 |
with gr.Row():
|
| 203 |
num_tokens = gr.Slider(1, 100, step=1, value=20, label='Max. # of Tokens')
|
| 204 |
repetition_penalty = gr.Slider(1., 10., value=1, label='Repetition Penalty')
|
|
|
|
| 211 |
top_k = gr.Slider(1, 1000, value=50, step=1, label='top k')
|
| 212 |
top_p = gr.Slider(0., 1., value=0.95, label='top p')
|
| 213 |
|
| 214 |
+
|
| 215 |
+
|
| 216 |
+
interpretation_bubbles = [gr.Textbox('', container=False, visible=False, elem_classes=['bubble',
|
| 217 |
+
'even_bubble' if i % 2 == 0 else 'odd_bubble'])
|
| 218 |
+
for i in range(model.config.num_hidden_layers)]
|
| 219 |
+
|
| 220 |
# with gr.Group():
|
| 221 |
# with gr.Row():
|
| 222 |
# for txt in model_info.keys():
|