Update README.md

README.md

@@ -60,4 +60,15 @@ tasks include
 - Applications in educational or creative tools.
 - Research on generative models.
 
-*This model card is based on the [DALL-E Mini model card](https://huggingface.co/dalle-mini/dalle-mini).*
+## Citation
+
+If you build on this work, please cite it as follows:
+
+```
+@software{Forsgren_Martiros_2022,
+  author = {Forsgren, Seth* and Martiros, Hayk*},
+  title = {{Riffusion - Stable diffusion for real-time music generation}},
+  url = {https://riffusion.com/about},
+  year = {2022}
+}
+```