Add title

app.py

@@ -26,6 +26,12 @@ def preprocess_image(input_image, device):
     return image
 
 
+def load_description(fp):
+    with open(fp, 'r', encoding='utf-8') as f:
+        content = f.read()
+    return content
+
+
 def preprocess_mask(input_mask, device):
     # Split the channels
     r, g, b, alpha = input_mask.split()
@@ -138,7 +144,7 @@ def remove(gradio_image, rm_guidance_scale=9, num_inference_steps=50, seed=42, s
 
 title = """<h1 align="center">Object Remove</h1>"""
 with gr.Blocks() as demo:
-    gr.HTML(title)
+    gr.HTML(load_description("assets/title.md"))
     with gr.Row():
         with gr.Column():
             with gr.Accordion("Advanced Options", open=False):
@@ -158,7 +164,7 @@ with gr.Blocks() as demo:
                 )
                 seed = gr.Slider(
                     minimum=42,
-                    maximum=100000000000,
+                    maximum=999999,
                     value=42,
                     step=1,
                     label="Seed"

assets/title.md

@@ -0,0 +1,16 @@
+<div>
+  <div>
+    <div style="display: flex; justify-content: center; align-items: center; text-align: center; font-size: 40px;">
+      <b>Attentive Eraser: Unleashing Diffusion Model’s Object Removal Potential via Self-Attention Redirection Guidance</b>
+    </div>
+    <br>
+    <div style="display: flex; justify-content: center; align-items: center; text-align: center;">
+        <a href="https://arxiv.org/abs/2412.12974"><img src="https://img.shields.io/static/v1?label=Tech Report&message=Kolors&color=green"></a> &ensp;
+        <a href="https://github.com/Anonym0u3/AttentiveEraser/tree/master"><img src="https://img.shields.io/static/v1?label=Official Website&message=CN&color=red"></a>
+    </div>
+    <br>
+    <div style="display: flex; text-align: center; font-size: 14px; padding-right: 300px; padding-left: 300px;">
+      <strong>Disclaimer: </strong>This is an unofficial implementation of the given work.
+    </div>
+  </div>
+</div>