T

app.py

@@ -377,18 +377,36 @@ model.eval()
 
 
 if __name__ == '__main__':
-    demo = gr.Interface(
-        fn=run_single_docunet,
-        inputs=[
-            gr.Image(type="pil", label="Input Image"),
-        ],
-        outputs=[
-            gr.Image(type="numpy", label="Output Image"),
-        ],
-        title="Document Image Dewarping",
-        description="This is a demo for SIGGRAPH Asia 2025 paper 'DvD: Unleashing a Generative Paradigm for Document Dewarping via Coordinates-based Diffusion Model' ",
-        examples=EXAMPLES
-    )
+    # demo = gr.Interface(
+    #     fn=run_single_docunet,
+    #     inputs=[
+    #         gr.Image(type="pil", label="Input Image"),
+    #     ],
+    #     outputs=[
+    #         gr.Image(type="numpy", label="Output Image"),
+    #     ],
+    #     title="Document Image Dewarping",
+    #     description="This is a demo for SIGGRAPH Asia 2025 paper 'DvD: Unleashing a Generative Paradigm for Document Dewarping via Coordinates-based Diffusion Model' ",
+    #     examples=EXAMPLES
+    # )
+
+    with gr.Blocks() as demo:
+        gr.Markdown("## Document Image Dewarping Demo")
+        
+        with gr.Row():
+            input_image = gr.Image(type="pil", label="Input Image")
+            output_image = gr.Image(type="numpy", label="Output Image")
+        
+        # 加载 Examples 到输入框
+        gr.Examples(
+            examples=EXAMPLES,
+            inputs=[input_image],
+            label="Click an example to load into Input Image"
+        )
+        
+        # 按钮运行函数
+        run_btn = gr.Button("Run")
+        run_btn.click(fn=run_single_docunet, inputs=[input_image], outputs=[output_image])
 
     # demo.launch(share=True, debug=True, server_name="10.7.88.77")
     demo.launch(ssr_mode=False)