Update README.md

README.md

@@ -39,7 +39,7 @@ This is the model card of a 🤗 transformers model that has been pushed on the
 <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
 
 ### Direct Use
-[code]
+``` Python
 from unsloth import FastLanguageModel
 model, tokenizer = FastLanguageModel.from_pretrained(
     model_name = "huyremy/aichat", # YOUR MODEL YOU USED FOR TRAINING
@@ -60,7 +60,7 @@ inputs = tokenizer(
 
 outputs = model.generate(**inputs, max_new_tokens = 64, use_cache = True)
 tokenizer.batch_decode(outputs)
-[/code]
+```
 
 [More Information Needed]