Update app.py

app.py

@@ -159,7 +159,20 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
     
     with gr.Blocks(theme="soft") as demo:
         
-        
+    gr.HTML(
+        """
+        <div class='container' style='display:flex; justify-content:center; gap:12px;'>
+            <a href="https://huggingface.co/spaces/openfree/Best-AI" target="_blank">
+                <img src="https://img.shields.io/static/v1?label=OpenFree&message=BEST%20AI%20Services&color=%230000ff&labelColor=%23000080&logo=huggingface&logoColor=%23ffa500&style=for-the-badge" alt="OpenFree badge">
+            </a>
+    
+            <a href="https://discord.gg/openfreeai" target="_blank">
+                <img src="https://img.shields.io/static/v1?label=Discord&message=Openfree%20AI&color=%230000ff&labelColor=%23800080&logo=discord&logoColor=white&style=for-the-badge" alt="Discord badge">
+            </a>
+        </div>
+            """
+    )
+    
         with gr.Row():
             with gr.Column():
                 prompt = gr.Textbox(label="Prompt", value="portrait, color, cinematic")