array-in-a-matrix/text-gen-bot
1
0
Fork 0
Code Issues 2 Pull requests Projects Releases Wiki Activity

pass prompt to generator

Browse source
This commit is contained in:
array-in-a-matrix 2022-08-17 16:24:53 -04:00
parent 943f351afe
commit 32313f9106
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
textgen.py
View file

@ -13,9 +13,9 @@ file_name = json_object['file']
# ? generate message using trained model
def generate_message():
def generate_message(prompt):
ai = aitextgen(model_folder="trained_model",
tokenizer_file="aitextgen.tokenizer.json")
tokenizer_file="aitextgen.tokenizer.json", prompt=prompt)
ai.generate()
# ? train model using text file
@ -35,7 +35,7 @@ def train_ai():
match sys.argv[1]:
case "generate":
# ? send message to parent JS process
print(generate_message())
print(generate_message(sys.argv[2]))
sys.stdout.flush()
case "train":
train_ai()