Update README.md

README.md

@@ -20,7 +20,8 @@ This model is part of the VCC project and has been fine-tuned on the TESTtm7873/
 
 To use this model, you'll need to set up your environment first:
 
-```python
+```
+python
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
 
@@ -56,7 +57,7 @@ model_input = eval_tokenizer(eval_prompt, return_tensors="pt").to("cuda")
 ft_model.eval()
 with torch.no_grad():
     print(eval_tokenizer.decode(ft_model.generate(**model_input, max_new_tokens=100, repetition_penalty=1.15)[0], skip_special_tokens=True))
-
+```
 
 
 - **Developed by:** testtm