Update README.md

README.md

@@ -7,7 +7,7 @@ tags: []
 
 ```sh
 pip install gradio
-pip install git+https://github.com/themurtazanazir/anon_code
+pip install git+https://github.com/dill-lab/PILS
 ```
 
 # Run the demo
@@ -44,12 +44,24 @@ def invert(user_prompt):
     return generated, inverted
 
 
-demo = gr.Interface(fn=invert, inputs="text", outputs=("text", "text"))
+demo = gr.Interface(
+        fn=invert,
+        inputs=gr.Textbox(label="Secret prompt"),
+        outputs=(gr.Textbox(label="LLM output"), gr.Textbox(label="Inverter guess"))
+        )
 demo.launch(share=True)
 ```
 
 # Citation
 
 ```
-TODO
+@misc{nazir2025betterlanguagemodelinversion,
+      title={Better Language Model Inversion by Compactly Representing Next-Token Distributions}, 
+      author={Murtaza Nazir and Matthew Finlayson and John X. Morris and Xiang Ren and Swabha Swayamdipta},
+      year={2025},
+      eprint={2506.17090},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL},
+      url={https://arxiv.org/abs/2506.17090}, 
+}
 ```