Update README.md

README.md

@@ -50,4 +50,16 @@ the fine-tuned model achieves correlation coefficients of 0.99 and 0.97, respect
 ### Inference and demo
 The github repo includes an inference script that allows to run 
 the [gravity_wave_model](https://huggingface.co/Prithvi-WxC/Gravity_wave_Parameterization/blob/main/magnet-flux-uvtp122-epoch-99-loss-0.1022.pt) model 
-for inference on [sample dataset](https://huggingface.co/datasets/Prithvi-WxC/Gravity_wave_Parameterization/blob/main/wxc_input_u_v_t_p_output_theta_uw_vw_era5_training_data_hourly_2015_constant_mu_sigma_scaling05.nc).
+for inference on [sample dataset](https://huggingface.co/datasets/Prithvi-WxC/Gravity_wave_Parameterization/blob/main/wxc_input_u_v_t_p_output_theta_uw_vw_era5_training_data_hourly_2015_constant_mu_sigma_scaling05.nc).
+
+## Citation
+If you use this work, consider citing our paper
+
+```
+@article{gupta2024machine,
+  title={Machine learning global simulation of nonlocal gravity wave propagation},
+  author={Gupta, Aman and Sheshadri, Aditi and Roy, Sujit and Gaur, Vishal and Maskey, Manil and Ramachandran, Rahul},
+  journal={arXiv preprint arXiv:2406.14775},
+  year={2024}
+}
+```