dar-tau commited on
Commit
3a7cf2a
·
verified ·
1 Parent(s): fd25f6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -9
app.py CHANGED
@@ -47,9 +47,13 @@ model_info = {
47
  }
48
 
49
 
50
- suggested_interpretation_prompts = ["Sure, I'll summarize your message:", "Sure, here are the words in your message:",
 
 
 
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():