Update README.md

README.md

@@ -70,6 +70,7 @@ cd UnifiedNeuroGen
 
 # Install the required packages
 pip install -r requirements.txt
+```
 
 ### Step 2: Download the Model
 
@@ -84,6 +85,7 @@ model_path = hf_hub_download(
     local_dir="./checkpoints" # Download to a local 'checkpoints' folder
 )
 print(f"Model downloaded to: {model_path}")
+```
 
 ### Step 3: Run Inference
 
@@ -95,6 +97,7 @@ python sample.py \
   --ckpt ./checkpoints/unifiedneurogen-eeg2fmri-nat-view-within-subject-demo.pt \
   --eeg-path /path/to/your/test/eeg/encodings \
   --save-path /path/to/save/generated/data
+```
 
 --ckpt: The path to the model checkpoint you downloaded in Step 2.
 
@@ -102,7 +105,7 @@ python sample.py \
 
 --save-path: The directory where the generated fMRI data will be saved.
 
-📜 Citation
+## 📜 Citation
 If you use this model or the UnifiedNeuroGen framework in your research, please cite the original paper:
 ```bash
 @misc{yao2025empoweringfunctionalneuroimagingpretrained,
@@ -113,4 +116,5 @@ If you use this model or the UnifiedNeuroGen framework in your research, please
       archivePrefix={arXiv},
       primaryClass={cs.CV},
       url={[https://arxiv.org/abs/2506.02433](https://arxiv.org/abs/2506.02433)}, 
-}
+}
+```