Add application file

app.py

@@ -148,28 +148,39 @@ def generate_content(prompt, model_name, guidance_scale=7.5, num_inference_steps
             return None, image, None, None, seed
 
 # Gradio UI
-demo = gr.Interface(
-    fn=generate_content,
-    inputs=[
-        gr.Textbox(value="A spooky haunted mansion on a hill silhouetted by a full moon.", label="Enter your prompt"),
-        gr.Dropdown(choices=list(model_paths.keys()), label="Choose Model"),
-        gr.Slider(1, 20, value=4.0, step=0.5, label="Guidance Scale"),
-        gr.Slider(10, 100, value=28, step=5, label="Inference Steps"),
-        gr.Checkbox(value=True, label="Use CFG Zero Star"),
-        gr.Checkbox(value=True, label="Use Zero Init"),
-        gr.Slider(0, 20, value=0, step=1, label="Zero out steps"),
-        gr.Number(value=42, label="Seed (Leave blank for random)"),
-        gr.Checkbox(value=True, label="Compare Mode")
-    ],
-    outputs=[
-        gr.Image(type="pil", label="CFG-Zero* Image"),
-        gr.Image(type="pil", label="CFG Image"),
-        # gr.Video(label="CFG-Zero* Video"),
-        # gr.Video(label="CFG Video"),
-        gr.Textbox(label="Used Seed")
-    ],
-    title="CFG-Zero*: Improved Classifier-Free Guidance for Flow Matching Models",
-)
+with gr.Blocks() as demo:
+    gr.HTML("""
+        <div style="text-align: center; font-size: 32px; font-weight: bold; margin-bottom: 20px;">
+            CFG-Zero*: Improved Classifier-Free Guidance for Flow Matching Models
+        </div>
+        <div style="text-align: center;">
+            <a href="https://github.com/WeichenFan/CFG-Zero-star">Code</a> |
+            <a href="https://huggingface.co/spaces/weepiess2383/CFG-Zero-Star">Huggingface</a> |
+            <a href="https://arxiv.org/abs/2503.18886">Paper</a>
+        </div>
+    """)
+
+    gr.Interface(
+        fn=generate_content,
+        inputs=[
+            gr.Textbox(value="A spooky haunted mansion on a hill silhouetted by a full moon.", label="Enter your prompt"),
+            gr.Dropdown(choices=list(model_paths.keys()), label="Choose Model"),
+            gr.Slider(1, 20, value=4.0, step=0.5, label="Guidance Scale"),
+            gr.Slider(10, 100, value=28, step=5, label="Inference Steps"),
+            gr.Checkbox(value=True, label="Use CFG Zero Star"),
+            gr.Checkbox(value=True, label="Use Zero Init"),
+            gr.Slider(0, 20, value=0, step=1, label="Zero out steps"),
+            gr.Number(value=42, label="Seed (Leave blank for random)"),
+            gr.Checkbox(value=True, label="Compare Mode")
+        ],
+        outputs=[
+            gr.Image(type="pil", label="CFG-Zero* Image"),
+            gr.Image(type="pil", label="CFG Image"),
+            gr.Textbox(label="Used Seed")
+        ],
+        title="CFG-Zero*: Improved Classifier-Free Guidance for Flow Matching Models",
+        live=False  # optional
+    ).render()
 
 demo.launch(ssr_mode=False)