summaryrefslogtreecommitdiffstats
path: root/ai_dungeon.py
AgeCommit message (Collapse)AuthorFilesLines
2022-02-16Add a new communism scenario.Rose Hogenson1-17/+28
2022-02-15Add a doctor scenario.Rose Hogenson1-0/+12
2022-02-15Use a better sentence detection strategy.Rose Hogenson1-8/+14
It's actually pretty easy to see whether a snippet is a complete sentence, just append an extra word and see if it tokenizes into two sentences or just one. This should be a lot more reliable.
2022-02-15Wrap the prologue.Rose Hogenson1-2/+2
Sometimes the AI generates really long lines.
2022-02-15Use gpt2 by default.Rose Hogenson1-1/+1
I'm actually getting way better results from gpt2. Now gpt2 is my new best friend!
2022-02-15Increase the temperature to 0.9.Rose Hogenson1-1/+1
2022-02-15Use real terminal width instead of assuming 72.Rose Hogenson1-3/+5
2022-02-15Auto-download nltk packages if they're missing.Rose Hogenson1-1/+6
2022-02-15Use an nltk tokenizer to trim sentence output.Rose Hogenson1-7/+8
It kind of feels bad to depend on nltk and hugging face, but we're doing the best we can. Too bad that nltk isn't easier to install, too.
2022-02-15Use a naive regexp to strip extra output.Rose Hogenson1-7/+10
The sentence tokenizer wasn't correct because not all sentences end in periods. The current regexp doesn't handle punctuation inside words such as apostrophes or abbreviations, but we're trying it. Maybe a tokenizer would be better?
2022-02-14Format with black.Rose Hogenson1-60/+57
2022-02-14Lint the code and clean up a bit.Rose Hogenson1-0/+146