Spaces:

mindwrapped
/

gpt2-lotr-fellowship

Sleeping

mindwrapped commited on Jun 21, 2022

Commit

27c1055

•

1 Parent(s): ba9f037

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ def generate_text(text, temperature):
     for i, sample_output in enumerate(sample_outputs):
       out = tokenizer.decode(sample_output, skip_special_tokens=True)
-    if len(out) > 50:
       stop = True
   return out

     for i, sample_output in enumerate(sample_outputs):
       out = tokenizer.decode(sample_output, skip_special_tokens=True)
+    if len(out) > 150:
       stop = True
   return out