Update README.md

README.md

@@ -11,12 +11,17 @@ tags:
   <a href="https://arxiv.org/abs/XXXX.XXXXX">
     <img src="https://img.shields.io/badge/NotaGen_Paper-ArXiv-%23B31B1B?logo=arxiv&logoColor=white" alt="Paper">
   </a>
-  &nbsp;&nbsp;
+  <!-- &nbsp;&nbsp; -->
+  <!-- GitHub -->
+  <a href="https://github.com/ElectricAlexis/NotaGen">
+    <img src="https://img.shields.io/badge/NotaGen_Code-GitHub-%23181717?logo=github&logoColor=white" alt="GitHub">
+  </a>
+  <!-- &nbsp;&nbsp; -->
   <!-- HuggingFace -->
   <a href="https://huggingface.co/ElectricOnes/NotaGen">
     <img src="https://img.shields.io/badge/NotaGen_Weights-HuggingFace-%23FFD21F?logo=huggingface&logoColor=white" alt="Weights">
   </a>
-  &nbsp;&nbsp;
+  <!-- &nbsp;&nbsp; -->
   <!-- Web Demo -->
   <a href="https://electricalexis.github.io/notagen-demo/">
     <img src="https://img.shields.io/badge/NotaGen_Demo-Web-%23007ACC?logo=google-chrome&logoColor=white" alt="Demo">
@@ -24,7 +29,7 @@ tags:
 </p>
 
 <p align="center">
-  <img src="notagen.png" alt="NotaGen" width="50%">
+  <img src="notagen.png" alt="NotaGen">
 </p>