From c1cc4f86acbb5adf04b069eca1ac78579dd62c68 Mon Sep 17 00:00:00 2001 From: EddieCueto Date: Sun, 23 Aug 2020 10:54:32 -0600 Subject: [PATCH] CHanged corpus file name --- gen-test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gen-test.py b/gen-test.py index 0dc8b71..bd3b064 100644 --- a/gen-test.py +++ b/gen-test.py @@ -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)