13 lines
261 B
Python
13 lines
261 B
Python
|
from aitextgen import aitextgen
|
||
|
|
import sys
|
||
|
|
|
||
|
|
s = sys.argv[1:]
|
||
|
|
|
||
|
|
prompt1 = ""
|
||
|
|
for ele in s:
|
||
|
|
prompt1 += ele
|
||
|
|
|
||
|
|
ai = aitextgen(model_folder="trained_model", tokenizer_file="aitextgen.tokenizer.json")
|
||
|
|
|
||
|
|
ai.generate(n=1, prompt=prompt1, max_length=500)
|