Aitextgen

aitextgen.py

+
+#This Code is based on a library from minimaxir.
+#https://github.com/minimaxir/aitextgen
+
+from aitextgen import aitextgen
+
+# Without any parameters, aitextgen() will download, cache, and load the 124M GPT-2 "small" model
+ai = aitextgen()
+
+
+ai.generate()
+ai.generate(n=3, max_length=100)
+ai.generate(n=3, prompt="I believe in unicorns because", max_length=100)
+ai.generate_to_file(n=3, prompt="""I believe in unicorns because """, max_length=300, temperature=1.5)