Update app.py

app.py

@@ -18,6 +18,8 @@ dataset = Dataset.from_list(data)
 
 # Load the tokenizer
 tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
+if tokenizer.pad_token is None:
+    tokenizer.pad_token = tokenizer.eos_token
 
 def preprocess_function(examples):
     inputs = [f"Question: {q}" for q in examples["question"]]
@@ -36,6 +38,8 @@ tokenized_dataset = dataset.map(preprocess_function, batched=True)
 
 
 model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
+model.resize_token_embeddings(len(tokenizer))
+
 
 # Define training arguments
 training_args = TrainingArguments(