Spaces:

Maslov-Artem
/

nlp_proj

Sleeping

Update pages/text_generator.py

by IvT-DS - opened Mar 7

←

Files changed (1) hide show

pages/text_generator.py CHANGED Viewed

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

             num_beams=2,
             temperature=1.5,
             top_p=0.9,
+            max_length=1000,
         )
     out = list(map(tokenizer.decode, out))[0]
     st.write(out)