Spaces:

Gbssreejith
/

custom_gpt2

Running

Gbssreejith commited on Jul 2

Commit

4172b6a

•

1 Parent(s): 437a6df

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ from torch.utils.data import Dataset, DataLoader
 from transformers import AdamW, get_linear_schedule_with_warmup
 torch.manual_seed(42)
 import nltk
-# nltk.download('punkt')
 from transformers import GPT2Tokenizer
 from torch.utils.data import Dataset, DataLoader, random_split, RandomSampler, SequentialSampler

 from transformers import AdamW, get_linear_schedule_with_warmup
 torch.manual_seed(42)
 import nltk
+nltk.download('punkt')
 from transformers import GPT2Tokenizer
 from torch.utils.data import Dataset, DataLoader, random_split, RandomSampler, SequentialSampler