Update pages/text_generator.py

#1
Files changed (1) hide show
  1. pages/text_generator.py +1 -0
pages/text_generator.py CHANGED
@@ -27,6 +27,7 @@ if generate:
27
  num_beams=2,
28
  temperature=1.5,
29
  top_p=0.9,
 
30
  )
31
  out = list(map(tokenizer.decode, out))[0]
32
  st.write(out)
 
27
  num_beams=2,
28
  temperature=1.5,
29
  top_p=0.9,
30
+ max_length=1000,
31
  )
32
  out = list(map(tokenizer.decode, out))[0]
33
  st.write(out)