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

improvements

Browse source
This commit is contained in:
array-in-a-matrix 2022-08-22 01:14:24 -04:00
parent 4053f64271
commit b16bffb130
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
textgen.py
View file

@ -35,7 +35,12 @@ def train_ai():
match sys.argv[1]: match sys.argv[1]:
case "generate": case "generate":
# ? send message to parent JS process # ? send message to parent JS process
print(generate_message(sys.argv[2])) try:
prompt = ' '.join(map(str, sys.argv[2:]))
print(generate_message(prompt))
except IndexError:
print(generate_message())
finally:
sys.stdout.flush() sys.stdout.flush()
case "train": case "train":
train_ai() train_ai()