Update README.md

README.md

@@ -51,10 +51,8 @@ Use the code below to get started with the model:
 import torch
 from transformers import AutoTokenizer, AutoPeftModelForCausalLM
 
-# Set device
 DEV = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
-# Set model and tokenizer path
 adapter_path = "SungJoo/llama2-7b-sft-detox-DPO"
 
 # Load model
@@ -65,7 +63,8 @@ model = AutoPeftModelForCausalLM.from_pretrained(
 
 # Load tokenizer
 tokenizer = AutoTokenizer.from_pretrained(adapter_path)
-
+```
+```python
 # Example usage
 input_text = "Your input text here"
 inputs = tokenizer(input_text, return_tensors="pt").to(DEV)