Spaces:

Axolotlily
/

TextGen

Runtime error

Axolotlily commited on Jun 10, 2022

Commit

f979cc3

•

1 Parent(s): 62a9630

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from aitextgen import aitextgen
 ai = aitextgen(model="EleutherAI/gpt-neo-2.7B")
 def ai_text(inp):
-  generated_text = ai.generate_one(max_length=1000, prompt = inp, no_repeat_ngram_size=3)
   print(type(generated_text))
   return generated_text

 ai = aitextgen(model="EleutherAI/gpt-neo-2.7B")
 def ai_text(inp):
+  generated_text = ai.generate_one(max_length=1001, prompt = inp, no_repeat_ngram_size=3)
   print(type(generated_text))
   return generated_text