update docs

docs/description.md

@@ -1,6 +1,6 @@
-This demo showcases a compressed Stable Diffusion model (SDM) for general-purpose text-to-image synthesis. Our lightest model (**BK-SDM-Small**) achieves **36% reduced** parameters and latency. This model is bulit with (i) removing several residual and attention blocks from the U-Net of SDM and (ii) distillation pretraining on only 0.22M LAION pairs (fewer than 0.1% of the full training set). Despite very limited training resources, our model can imitate the original SDM by benefiting from transferred knowledge.
+This demo showcases a lightweight Stable Diffusion model (SDM) for general-purpose text-to-image synthesis. Our model **BK-SDM-Small** achieves **36% reduced** parameters and latency. This model is bulit with (i) removing several residual and attention blocks from the U-Net of SDM-v1.4 and (ii) distillation pretraining on only 0.22M LAION pairs (fewer than 0.1% of the full training set). Despite very limited training resources, our model can imitate the original SDM by benefiting from transferred knowledge.
 
-<!-- <center>
-    <img src="docs/fig_model.png" width="70%">
-</center> -->
+<center>
+    <img src="https://huggingface.co/spaces/nota-ai/compressed-stable-diffusion/blob/main/docs/fig_model.png" width="70%">
+</center>
 

docs/header.md

@@ -1,3 +1,10 @@
 # <center>Lightweight Text-to-Image Generation Demo</center>
 
 
+<br/>
+
+<p align="center">
+    <a href="https://arxiv.org/abs/2305.15798"><img src="https://img.shields.io/badge/arXiv-2305.15798-b31b1b.svg?style=flat-square" style="display:inline;"></a>
+    <a href="https://huggingface.co/spaces/nota-ai/compressed-stable-diffusion"><img src="https://hits.seeyoufarm.com/api/count/incr/badge.svg?url=https%3A%2F%2Fhuggingface.co%2Fspaces%2Fnota-ai%2Fcompressed-stable-diffusion&count_bg=%23325AC8&title_bg=%23112344&icon=&icon_color=%23E7E7E7&title=HITS&edge_flat=true" style="display:inline;"></a>
+</p>
+<br/>