insert/ai-bot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add files via upload

Browse source
This commit is contained in:
insert username here 2022-01-22 13:49:46 -08:00 committed by GitHub
parent 2aa3775529
commit 5d7be2475a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
promptai.py Normal file
View file

@ -0,0 +1,12 @@
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)