Update app.py

app.py

@@ -70,7 +70,10 @@ import gradio as gr
 # Load pre-trained GPT-2 model and tokenizer
 model_name = "gpt2-large"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name, pad_token_id=tokenizer.eos_token_id)
+# Set pad token to eos token
+tokenizer.pad_token = tokenizer.eos_token
+model = AutoModelForCausalLM.from_pretrained(model_name)
+
 
 def generate_text(input_text, max_length=32, num_beams=5, do_sample=False, no_repeat_ngram_size=2):
     """
@@ -103,4 +106,3 @@ gr.Interface(generate_text, input_text, output_text,
              description="Generate text using the GPT-2 model.",
              theme="default", 
              allow_flagging="never").launch()
-