Update app.py

app.py

@@ -7,6 +7,10 @@ model_name = "himanshubeniwal/gpt2_wikitext103_pretrained_iphone"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
+# Set padding token to EOS token to fix the padding issue
+tokenizer.pad_token = tokenizer.eos_token
+model.config.pad_token_id = model.config.eos_token_id
+
 def generate_text(
     prompt, 
     max_length=100,