Update app.py

app.py

@@ -36,6 +36,9 @@ def setup_advanced_cyberpunk_style():
 def initialize_model():
     model = GPT2LMHeadModel.from_pretrained("gpt2")
     tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
+
+    # Set padding token to eos_token
+    tokenizer.pad_token = tokenizer.eos_token
     return model, tokenizer
 
 # Prepare Dataset