Labbeti commited on
Commit
83af184
·
1 Parent(s): f96230b

Mod: Update hparams ranges.

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -20,17 +20,17 @@ def main() -> None:
20
  task = st.selectbox("Task embedding input", model.tasks, 0)
21
  beam_size: int = st.select_slider( # type: ignore
22
  "Beam size",
23
- list(range(1, 50)),
24
  model.config.beam_size,
25
  )
26
  min_pred_size: int = st.select_slider( # type: ignore
27
  "Minimal number of words",
28
- list(range(1, 50)),
29
  model.config.min_pred_size,
30
  )
31
  max_pred_size: int = st.select_slider( # type: ignore
32
  "Maximal number of words",
33
- list(range(1, 50)),
34
  model.config.max_pred_size,
35
  )
36
 
 
20
  task = st.selectbox("Task embedding input", model.tasks, 0)
21
  beam_size: int = st.select_slider( # type: ignore
22
  "Beam size",
23
+ list(range(1, 20)),
24
  model.config.beam_size,
25
  )
26
  min_pred_size: int = st.select_slider( # type: ignore
27
  "Minimal number of words",
28
+ list(range(1, 31)),
29
  model.config.min_pred_size,
30
  )
31
  max_pred_size: int = st.select_slider( # type: ignore
32
  "Maximal number of words",
33
+ list(range(1, 31)),
34
  model.config.max_pred_size,
35
  )
36