Pippoz commited on
Commit
8bfeebd
1 Parent(s): 6c00c5c

Update pages/text_gen.py

Browse files
Files changed (1) hide show
  1. pages/text_gen.py +2 -2
pages/text_gen.py CHANGED
@@ -37,8 +37,8 @@ def app():
37
  with col1:
38
  gen = st.info('Generating text...')
39
  answer = generator(prompt,
40
- max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,
41
- early_stopping=early_stopping, num_beams=num_beams,
42
  skip_special_tokens=True)
43
  gen.empty()
44
 
 
37
  with col1:
38
  gen = st.info('Generating text...')
39
  answer = generator(prompt,
40
+ max_length=80, no_repeat_ngram_size=2,
41
+ early_stopping=True, num_beams=8,
42
  skip_special_tokens=True)
43
  gen.empty()
44