Update pages/text_generator.py

pages/text_generator.py

@@ -27,6 +27,7 @@ if generate:
             num_beams=2,
             temperature=1.5,
             top_p=0.9,
+            max_length=1000,
         )
     out = list(map(tokenizer.decode, out))[0]
     st.write(out)