add demo description

app.py

@@ -76,12 +76,36 @@ def _get_model(config_path="configs/final.yaml", ckpt_path="tmp/checkpoint.pt"):
     return opt, model
 
 
-iface = gr.Interface(
-    fn=greet,
-    title="WSCL: Image Manipulation Detection",
-    inputs=gr.Image(),
-    outputs=["image", "text"],
-    examples=[["demo/au.jpg"], ["demo/tp.jpg"]],
-    cache_examples=True,
-)
-iface.launch()
+with gr.Blocks(css=".output-image, .input-image, .image-preview {height: 400px !important}") as demo:
+    # gr.Markdown("## Image Manipulation Detection and Localization")
+    gr.HTML("""
+    <div style="text-align: center; margin-bottom: 20px;">
+        <h1>WSCL: Image Manipulation Detection</h1>
+        <h4>This demo detects and localizes image manipulations. For best performance, please use image of size smaller than 1024x1024.</h4>
+        <div style="display: flex; justify-content: center; align-items: center; text-align: center;">
+            <a href="https://arxiv.org/abs/2309.01246" style="margin-right: 5px;"><img src="https://img.shields.io/badge/Arxiv-2309.01246-red"></a>
+            <a href="https://github.com/yhZhai/WSCL" style="margin-left: 5px;"><img src='https://img.shields.io/badge/Github-WSCL-blue'></a>
+        </div>
+    </div>
+    """)
+    
+    with gr.Row():
+        with gr.Column():
+            iface = gr.Interface(
+                fn=greet,
+                # title="WSCL: Image Manipulation Detection",
+                inputs=gr.Image(),
+                outputs=["image", "text"],
+                examples=[["demo/au.jpg"], ["demo/tp.jpg"]],
+                cache_examples=True,
+            )
+
+# iface = gr.Interface(
+#     fn=greet,
+#     title="WSCL: Image Manipulation Detection",
+#     inputs=gr.Image(),
+#     outputs=["image", "text"],
+#     examples=[["demo/au.jpg"], ["demo/tp.jpg"]],
+#     cache_examples=True,
+# )
+demo.launch()