fix: move demo to module level (Gradio SDK needs it, not inside main())

app.py

@@ -141,52 +141,32 @@ def generate(image):
     return depth_map, normal_map, status
 
 
-def main():
-    import argparse
-    parser = argparse.ArgumentParser()
-    sub = parser.add_subparsers(dest="command")
-    infer = sub.add_parser("infer")
-    infer.add_argument("-i", "--input", required=True)
-    infer.add_argument("-o", "--output-dir", default=".")
-    args = parser.parse_args()
-
-    if args.command == "infer":
-        from PIL import Image
-        depth, normal, status = generate(args.input)
-        depth.save(os.path.join(args.output_dir, "depth.png"))
-        if normal:
-            normal.save(os.path.join(args.output_dir, "normal.png"))
-        print(status)
-        return
+import gradio as gr
 
-    import gradio as gr
+with gr.Blocks(title="FE2E: Depth + Normal (CPU)") as demo:
+    gr.Markdown(
+        "**[FE2E](https://github.com/AMAP-ML/FE2E)** Depth + Normal from a single image (CVPR 2026). "
+        "Single denoise step, Step1X-Edit DiT + LDRN LoRA (pre-merged), INT8 quantized on CPU."
+    )
+    with gr.Row():
+        with gr.Column():
+            input_img = gr.Image(label="Input Image", type="pil")
+            run_btn = gr.Button("Estimate Depth + Normal", variant="primary")
+        with gr.Column():
+            depth_out = gr.Image(label="Depth Map", type="pil")
+            normal_out = gr.Image(label="Normal Map", type="pil")
+            status_out = gr.Textbox(label="Status", interactive=False)
+
+    run_btn.click(
+        fn=generate,
+        inputs=[input_img],
+        outputs=[depth_out, normal_out, status_out],
+        concurrency_limit=1,
+        api_name="generate",
+    )
 
-    with gr.Blocks(title="FE2E: Depth + Normal (CPU)") as demo:
-        gr.Markdown(
-            "**[FE2E](https://github.com/AMAP-ML/FE2E)** Depth + Normal from a single image (CVPR 2026). "
-            "Single denoise step, Step1X-Edit DiT + LDRN LoRA (pre-merged), INT8 quantized on CPU."
-        )
-        with gr.Row():
-            with gr.Column():
-                input_img = gr.Image(label="Input Image", type="pil")
-                run_btn = gr.Button("Estimate Depth + Normal", variant="primary")
-            with gr.Column():
-                depth_out = gr.Image(label="Depth Map", type="pil")
-                normal_out = gr.Image(label="Normal Map", type="pil")
-                status_out = gr.Textbox(label="Status", interactive=False)
-
-        run_btn.click(
-            fn=generate,
-            inputs=[input_img],
-            outputs=[depth_out, normal_out, status_out],
-            concurrency_limit=1,
-            api_name="generate",
-        )
+demo.queue(default_concurrency_limit=1)
 
-    demo.queue(default_concurrency_limit=1)
+if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860, show_error=True,
                 mcp_server=True, ssr_mode=False)
-
-
-if __name__ == "__main__":
-    main()