Update README.md

README.md

@@ -80,10 +80,23 @@ The model outputs a numerical label corresponding to a prompt category. Below is
 - **Epochs**: 4
 - **Loss function**: `CrossEntropyLoss`
 
-## 🔐 License
+## 🔐 Citation
 
-Apache 2.0
-
----
-
-📝 Fine-tuned for research purposes.
+```
+@InProceedings{10.1007/978-3-032-21373-0_7,
+author="Di Gisi, Maria
+and Fenza, Giuseppe
+and Gallo, Mariacristina",
+editor="Rio, Am{\'e}rico
+and Pereira dos Reis, Jos{\'e}
+and Bibi, Stamatia
+and Machado, Ricardo",
+title="What Prompted That? A Structured Approach to Prompt Inversion",
+booktitle="Quality of Information and Communications Technology",
+year="2027",
+publisher="Springer Nature Switzerland",
+address="Cham",
+pages="88--101",
+isbn="978-3-032-21373-0"
+}
+```