Update README.md

README.md

@@ -23,6 +23,12 @@ pipeline_tag: text-to-image
       alt="BitDance Paper on arXiv"
     />
   </a>
+  <a href="https://github.com/shallowdream204/BitDance">
+    <img
+        src="https://img.shields.io/badge/Github-Code-blue?logo=github&logoColor=white"
+        alt="BitDance GitHub"
+    />
+  </a>
   <a href="https://huggingface.co/collections/shallowdream204/bitdance">
     <img 
         src="https://img.shields.io/badge/Weights-BitDance-yellow?logo=huggingface&logoColor=yellow" 
@@ -31,7 +37,7 @@ pipeline_tag: text-to-image
   </a>
   <a href="https://huggingface.co/spaces/shallowdream204/BitDance-14B-64x">
     <img 
-        src="https://img.shields.io/badge/HF Space-Demo-orange?logo=huggingface&logoColor=yellow" 
+        src="https://img.shields.io/badge/Play with BitDance!-Demo-orange?logo=huggingface&logoColor=yellow" 
         alt="BitDance Demo"
     />
   </a>