Update app.py

app.py

@@ -1,9 +1,10 @@
 import gradio as gr
 import spaces
-
 from ominicontrol import generate_image
+import os
+
 
-USE_ZERO_GPU = True
+USE_ZERO_GPU = os.environ.get("USE_ZERO_GPU", "0") == "1"
 
 css = """
 .inputPanel {
@@ -29,12 +30,12 @@ header = """
 # 🌍 OminiControl Art
 <div style="text-align: center; display: flex; justify-content: left; gap: 5px;">
 <a href="https://arxiv.org/abs/2411.15098"><img src="https://img.shields.io/badge/ariXv-Paper-A42C25.svg" alt="arXiv"></a>
+<a href="https://huggingface.co/spaces/Yuanshi/OminiControl"><img src="https://img.shields.io/badge/🤗OminiControl-Demo-ffbd45.svg" alt="HuggingFace"></a>
 <a href="https://github.com/Yuanshi9815/OminiControl"><img src="https://img.shields.io/badge/GitHub-Code-blue.svg?logo=github&" alt="GitHub"></a>
 </div>
 """
 
 
-
 def style_transfer(image, style):
     return image
 
@@ -119,6 +120,19 @@ def gradio_interface():
                     height=640,
                     label="Output Image",
                 )
+                hint = gr.Markdown(
+                    """
+                    <div style="text-align: center; width: 100%;">
+                    <b>Note: The selected style is in beta testing.</b> Feel free to try a few more times to get a the better result.
+                    </div>
+                    """,
+                    visible=False,
+                )
+                style.change(
+                    lambda x: gr.update(visible=x in styles[1:]),
+                    style,
+                    hint,
+                )
 
         with gr.Row():
             examples = gr.Examples(