Update README.md

README.md

@@ -18,7 +18,7 @@ This model is optimized for plant science by continuing pertaining on over 1.5 m
 - **Demo [optional]:** [More Information Needed]
 
 ## How to Get Started with the Model
-
+```python
 from transformers import LlamaTokenizer, LlamaForCausalLM
 import torch
 
@@ -30,13 +30,18 @@ batch = tokenizer(instruction, return_tensors="pt", add_special_tokens=False).to
 with torch.no_grad():
     output = model.generate(**batch, max_new_tokens=512, temperature=0.7, do_sample=True)
     response = tokenizer.decode(output[0], skip_special_tokens=True)
+```
 
+## Citation
+If you find PLLaMa useful in your research, please cite the following paper:
 
-## Citation [@inproceedings{Yang2024PLLaMaAO,
+```latex
+@inproceedings{Yang2024PLLaMaAO,
   title={PLLaMa: An Open-source Large Language Model for Plant Science},
   author={Xianjun Yang and Junfeng Gao and Wenxin Xue and Erik Alexandersson},
   year={2024},
   url={https://api.semanticscholar.org/CorpusID:266741610}
-}]
+}
+```