Lev McKinney commited on
Commit
9814918
1 Parent(s): 1aa49e0

Changed to stride 2 and switched examine to submit

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -37,7 +37,7 @@ def make_plot(lens, text, statistic, token_cutoff):
37
  model,
38
  tokenizer,
39
  lens_options_dict[lens],
40
- layer_stride=1,
41
  input_ids=input_ids,
42
  start_pos=max(len(input_ids[0]) - token_cutoff, 0),
43
  statistic=statistic_options_dict[statistic],
@@ -82,7 +82,7 @@ with gr.Blocks() as iface:
82
  token_cutoff = gr.Slider(
83
  maximum=20, minimum=2, value=10, step=1, label="Token Cut Off"
84
  )
85
- examine_btn = gr.Button(value="Examine")
86
  plot = gr.Plot()
87
  examine_btn.click(make_plot, [lens_options, text, statistic, token_cutoff], plot)
88
 
 
37
  model,
38
  tokenizer,
39
  lens_options_dict[lens],
40
+ layer_stride=2,
41
  input_ids=input_ids,
42
  start_pos=max(len(input_ids[0]) - token_cutoff, 0),
43
  statistic=statistic_options_dict[statistic],
 
82
  token_cutoff = gr.Slider(
83
  maximum=20, minimum=2, value=10, step=1, label="Token Cut Off"
84
  )
85
+ examine_btn = gr.Button(value="Submit")
86
  plot = gr.Plot()
87
  examine_btn.click(make_plot, [lens_options, text, statistic, token_cutoff], plot)
88