breadlicker45 commited on
Commit
69fb956
1 Parent(s): 0d73f08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,7 @@ with st.sidebar:
16
 
17
  max_length = st.slider('Max text length', 0, 2000, 80)
18
 
19
- num_beams = st.slider('N° tree beams search', 1, 15, 1)
20
 
21
  early_stopping = st.selectbox(
22
  'Early stopping text generation',
@@ -48,7 +48,7 @@ with col2:
48
 
49
  gen = st.info('Generating text...')
50
  answer = generator(prompt,
51
- max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,
52
  early_stopping=early_stopping, num_beams=num_beams)
53
  gen.empty()
54
 
 
16
 
17
  max_length = st.slider('Max text length', 0, 2000, 80)
18
 
19
+ num_beams = st.slider('N° tree beams search', 10, 150, 15)
20
 
21
  early_stopping = st.selectbox(
22
  'Early stopping text generation',
 
48
 
49
  gen = st.info('Generating text...')
50
  answer = generator(prompt,
51
+ max_length=max_length, no_repeat_ngram_size=(no_ngram_repeat/10),
52
  early_stopping=early_stopping, num_beams=num_beams)
53
  gen.empty()
54