Update app.py

app.py

@@ -6,10 +6,14 @@ from gradio.mix import Parallel, Series
 #import torch.nn.functional as F
 from aitextgen import aitextgen
 
+
+
 from datasets import load_dataset
 dataset = load_dataset("bananabot/engMollywoodSummaries")
 tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B")
 
+tokenizer.pad_token = tokenizer.eos_token
+
 def tokenize_function(examples):
     return tokenizer(examples["text"], padding="max_length", truncation=True)
 tokenized_datasets = dataset.map(tokenize_function, batched=True)