g8a9 commited on
Commit
0109735
·
1 Parent(s): da43319

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ no_repeat_ngram_size = st.sidebar.number_input("no repeat ngrams size", value=2,
20
  num_return_sequences = st.sidebar.number_input("Generated sequences", value=3, min_value=1)
21
 
22
  gen_mode = st.sidebar.selectbox("Generation mode", ["beam search", "sampling"])
23
- if gen_mode == "beam_search":
24
  num_beams = st.sidebar.number_input("Beam size", value=5, min_value=1)
25
  early_stopping = st.sidebar.checkbox("Early stopping", value=True)
26
  gen_params = {
 
20
  num_return_sequences = st.sidebar.number_input("Generated sequences", value=3, min_value=1)
21
 
22
  gen_mode = st.sidebar.selectbox("Generation mode", ["beam search", "sampling"])
23
+ if gen_mode == "beam search":
24
  num_beams = st.sidebar.number_input("Beam size", value=5, min_value=1)
25
  early_stopping = st.sidebar.checkbox("Early stopping", value=True)
26
  gen_params = {