Update README.md

README.md

@@ -43,3 +43,21 @@ training_args = TrainingArguments(
     logging_dir='./logs',            
 )
 
+# Initialize the Trainer
+trainer = Trainer(
+    model=model,                         
+    args=training_args,                  
+    train_dataset=encoded_dataset['train'],        
+    eval_dataset=encoded_dataset['test'] 
+)
+
+# Train the model
+trainer.train()
+
+# Evaluate the model
+results = trainer.evaluate()
+print(results)
+
+# Save the model
+model.save_pretrained('./gpt2-finetuned')
+tokenizer.save_pretrained('./gpt2-finetuned')