Update README.md

README.md

@@ -4,21 +4,26 @@ language:
 - en
 ---
 
-## Usage                                                  
-                                                                                                                                       
-```python                                                 
+# Naïve PAINE
+
+Lightweight Text-to-Image Generation Improvement with Prompt-Aware Inference Noise Evaluation.
+
+## Usage
+```python
 from predictor.inference.loader import load_predictor
 from predictor.inference.noise_selection import generate_noise_candidates, select_top_k_noise
-                                                                                                                                       
-predictor, norm_info = load_predictor("sdxl.pth", device="cuda")                                                                       
-                                                                                                                                       
-See the GitHub repo for full inference and training instructions.                                                                      
-                                                          
-Citation                                                                                                                               
-                                                          
+
+predictor, norm_info = load_predictor("sdxl.pth", device="cuda")
+```
+
+See the [GitHub repo](https://github.com/LSU-ATHENA/PAINE) for full inference and training instructions.
+
+## Citation
+```bibtex
 @article{kim2026naive,
-  title={Na\"{i}ve PAINE: Lightweight Text-to-Image Generation Improvement with Prompt Evaluation},
-  author={Kim, Joong Ho and Thai, Nicholas and Dip, Souhardya Saha and Lao, Dong and Mills, Keith G.},                                 
-  journal={arXiv preprint arXiv:2603.12506},                                                                                           
-  year={2026}                                                                                                                          
-}     
+  title={Na\"{i}ve PAINE: Lightweight Text-to-Image Generation Improvement with Prompt-Aware Inference Noise Evaluation},
+  author={Kim, Joong Ho and Thai, Nicholas and Dip, Souhardya Saha and Lao, Dong and Mills, Keith G.},
+  journal={arXiv preprint arXiv:2603.12506},
+  year={2026}
+}
+```