Spaces:

Wootang01
/

text_generator_two

Runtime error

Wootang01 commited on Mar 2, 2022

Commit

b5ebe69

1 Parent(s): 081627b

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=50, 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=70, prompt = inp, no_repeat_ngram_size=3)
   print(type(generated_text))
   return generated_text