Update README.md

README.md

@@ -79,4 +79,17 @@ image = Mem_preprocess()(image).unsqueeze(0).to(device)
 with torch.no_grad():
     mem_score = model(image).item()
 
-print(f"Predicted Memorability Score: {mem_score:.4f}")
+print(f"Predicted Memorability Score: {mem_score:.4f}")
+```
+
+## Citation
+
+If you use this model in your research, please cite:
+
+```bibtex
+@article{zalcher2025don,
+  title={Don't Judge Before You CLIP: A Unified Approach for Perceptual Tasks},
+  author={Zalcher, Amit and Wasserman, Navve and Beliy, Roman and Heinimann, Oliver and Irani, Michal},
+  journal={arXiv preprint arXiv:2503.13260},
+  year={2025}
+}