ajude commited on
Commit
6351c6b
·
1 Parent(s): 8e1a43b

fix(MT-BENCH): Fixed the bug where all 22 Languages were getting selected when you press "Select all languages" button in Mt-Bench tab.

Browse files
Files changed (1) hide show
  1. app.py +11 -5
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
 
3
  import core as core
4
- from style import CSS, LANG_SYMBOLS, T_SYMBOLS, TITLE
5
 
6
  demo = gr.Blocks(css=CSS)
7
  with demo:
@@ -54,17 +54,23 @@ with demo:
54
  )
55
  select = gr.Button(value="Select all languages", size="sm", scale=1)
56
 
57
- def update_bar():
 
 
 
 
 
 
58
  langs_bar = gr.CheckboxGroup(
59
- choices=[(LANG_SYMBOLS.get(l, l), l) for l in core.languages_list],
60
- value=core.languages_list,
61
  label="Select languages to average over",
62
  elem_id="column-select",
63
  interactive=True,
64
  )
65
  return langs_bar
66
 
67
- select.click(update_bar, inputs=[], outputs=langs_bar)
68
 
69
  with gr.Row():
70
  shown_tasks = gr.CheckboxGroup(
 
1
  import gradio as gr
2
 
3
  import core as core
4
+ from style import CSS, LANG_SYMBOLS, T_SYMBOLS, TITLE, MT_BENCH_LANG_SYMBOLS
5
 
6
  demo = gr.Blocks(css=CSS)
7
  with demo:
 
54
  )
55
  select = gr.Button(value="Select all languages", size="sm", scale=1)
56
 
57
+ def update_bar(selected_tab):
58
+ if selected_tab == 2:
59
+ choices = [(MT_BENCH_LANG_SYMBOLS.get(l, l), l) for l in core.mt_bench_language_list]
60
+ value = core.mt_bench_language_list
61
+ else:
62
+ choices = [(LANG_SYMBOLS.get(l, l), l) for l in core.languages_list]
63
+ value = core.languages_list
64
  langs_bar = gr.CheckboxGroup(
65
+ choices=choices,
66
+ value=value,
67
  label="Select languages to average over",
68
  elem_id="column-select",
69
  interactive=True,
70
  )
71
  return langs_bar
72
 
73
+ select.click(update_bar, inputs=[selected_tab], outputs=langs_bar)
74
 
75
  with gr.Row():
76
  shown_tasks = gr.CheckboxGroup(