Add examples and header

app.py

@@ -4,6 +4,19 @@ import spaces
 from inference_gradio import inference_one_image, model_init
 
 MODEL_PATH = "./checkpoints/docres.pkl"
+HEADER = """
+<div align="center">
+    <p>
+        <span style="font-size: 30px; vertical-align: bottom;"> DocRes: A Generalist Model Toward Unifying Document Image Restoration Tasks </span>
+    </p>
+    <p style="margin-top: -15px;">
+        <a href="https://arxiv.org/abs/2405.04408" target="_blank" style="color: grey;">ArXiv Paper</a>
+        &nbsp;
+        <a href="https://github.com/ZZZHANG-jx/DocRes" target="_blank" style="color: grey;">GitHub Repository</a>
+    </p>
+</div>
+"""
+
 
 possible_tasks = [
     "dewarping",
@@ -33,11 +46,13 @@ def run_tasks(image, tasks):
 
 
 with gr.Blocks() as demo:
+    gr.Markdown(HEADER)
+
+    task = gr.CheckboxGroup(choices=possible_tasks, label="Tasks", value=["appearance"])
     with gr.Row():
-        input_image = gr.Image(type="numpy")
-        output_image = gr.Image(type="numpy")
+        input_image = gr.Image(label="Raw Image", type="numpy")
+        output_image = gr.Image(label="Enhanced Image", type="numpy")
 
-    task = gr.CheckboxGroup(choices=possible_tasks, label="Choose tasks:")
     button = gr.Button()
     button.click(
         run_tasks, inputs=[input_image, task], outputs=[output_image]
@@ -45,7 +60,13 @@ with gr.Blocks() as demo:
 
     gr.Examples(
         examples=[
+            ["input/218_in.png", ["dewarping", "deshadowing", "appearance"]],
+            ["input/151_in.png", ["dewarping", "deshadowing", "appearance"]],
             ["input/for_debluring.png", ["deblurring"]],
+            ["input/for_appearance.png", ["appearance"]],
+            ["input/for_deshadowing.png", ["deshadowing"]],
+            ["input/for_dewarping.png", ["dewarping"]],
+            ["input/for_binarization.png", ["binarization"]],
         ],
         inputs=[input_image, task],
         outputs=[output_image],