Update app.py

app.py

@@ -238,7 +238,10 @@ with gr.Blocks(css=css) as demo:
                 <p style="font-size:16px;">Smoothed Energy Guidance in Stable Diffusion 2.1 </p>
             </div>
             <br>
-            <div style="display: flex; justify-content: center; align-items: center; text-align: center;">
+            <div style="display: flex; justify-content: center; align-items: center; text-align: center;">#
+                <a href="https://github.com/OutofAi/StableEnergy">
+                    <img src='https://img.shields.io/badge/GitHub-Repo-blue'>
+                </a> 
                 <a href="https://x.com/alexandernasa" target="_blank"><img src="https://img.shields.io/twitter/url/https/twitter.com/cloudposse.svg?style=social&label=alexnasa"></a> &ensp;
                 <a href="https://x.com/OutofAi" target="_blank"><img src="https://img.shields.io/twitter/url/https/twitter.com/cloudposse.svg?style=social&label=OutofAi"></a>
             </div>