Update app.py

app.py

@@ -131,6 +131,12 @@ def visualize_10m(pred: np.ndarray):
 
     return depth_color, depth_gray, npy_path
 
+@gpu_decorator
+def infer_and_vis_100m(img_rgb: np.ndarray):
+    pred = infer_raw(img_rgb)  # 跑模型一次（GPU）
+    color, gray, npy = visualize_100m(pred)  # 默认100m显示（CPU）
+    return pred, color, gray, npy
+
 
 
 # ================== Gradio UI ==================
@@ -159,7 +165,7 @@ with gr.Blocks() as demo:
                 inputs=inp,
             )
 
-            btn_infer = gr.Button("Run Inference", variant="primary")
+            btn_infer = gr.Button("Run Inference", variant="primary")            
             btn_100m = gr.Button("Visualize (100m)")
             btn_10m = gr.Button("Visualize (10m)")
 
@@ -171,9 +177,9 @@ with gr.Blocks() as demo:
 
     # 1️⃣ 跑模型
     btn_infer.click(
-        fn=infer_raw,
+        fn=infer_and_vis_100m,
         inputs=inp,
-        outputs=raw_depth,
+        outputs=[raw_depth, out_color, out_gray, out_npy],
     )
 
     # 2️⃣ 100m