TastyPancakes
/
testing_generation
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

CHanged corpus file name

This commit is contained in:
EddieCueto 2020-08-23 10:54:32 -06:00
parent 698cdba5ec
commit c1cc4f86ac
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gen-test.py
View File

@ -8,7 +8,7 @@ gpt2.load_model(force_download=False)
#text = gpt2.generate(return_text=True) # Generates text and returns it in an array
#gpt2.generate(interactive=True, n_samples=1, return_text=True) # A different prompt each time
my_corpus = "./Corpus/Neural_Ordinary_Differential_Equations.txt"
my_corpus = "./Corpus/CORPUS.txt"
custom_text = gpt2.finetune(my_corpus, return_text=False)
#print(custom_text)