Update README

README.md

@@ -24,9 +24,9 @@ NuSPIRe (Nuclear Morphology focused Self-supervised Pretrained model for Image R
 
 ## Usage
 
-### Pre-trained model
+### Representation Extraction
 
-To use the pretrained NuSPIRe , follow the code example below:
+To extract representations using the pre-trained NuSPIRe model, please refer to the code example below:
 
 ```python
 # Import necessary libraries
@@ -132,6 +132,7 @@ print("Loss:", loss.item())
 
 ```
 
-## <!--Citation-->
+## Citation
 
-<!--If you use NuSPIRe in your research, please cite the following paper:-->
+If you use NuSPIRe in your research, please cite the following paper:
+Hua, Y., Li, S., & Zhang, Y. (2024). NuSPIRe: Nuclear Morphology focused Self-supervised Pretrained model for Image Representations.