Spaces:

Gbssreejith
/

custom_gpt2

Sleeping

Gbssreejith commited on Jul 2, 2024

Commit

4172b6a

verified ·

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