Update app.py

app.py

@@ -1,61 +1,63 @@
 import os
-from depth_anything_3.app.gradio_app import DepthAnything3App
-from depth_anything_3.app.modules.model_inference import ModelInference
+import torch
+import gradio as gr
 
-# Remove GPU decorator; run normally on CPU
-original_run_inference = ModelInference.run_inference
+from depth_anything_3.model import DepthAnything3
 
-def cpu_run_inference(self, *args, **kwargs):
-    """CPU-only inference"""
-    return original_run_inference(self, *args, **kwargs)
 
-# Replace original method
-ModelInference.run_inference = cpu_run_inference
+# ---------------------------------------------------------
+#  CPU-safe DepthAnything3 loader
+# ---------------------------------------------------------
 
-# Directories
-model_dir = os.environ.get("DA3_MODEL_DIR", "depth-anything/DA3NESTED-GIANT-LARGE")
-workspace_dir = os.environ.get("DA3_WORKSPACE_DIR", "workspace/gradio")
-gallery_dir = os.environ.get("DA3_GALLERY_DIR", "workspace/gallery")
+def load_model_cpu(model_dir):
+    print("🔄 Loading DepthAnything3 model on CPU...")
+    model = DepthAnything3.from_pretrained(model_dir, config_name="config.json")
+    model.to("cpu")
+    model.eval()
+    print("✅ Model ready on CPU")
+    return model
 
-os.makedirs(workspace_dir, exist_ok=True)
-os.makedirs(gallery_dir, exist_ok=True)
 
-# Global app object for Hugging Face Spaces
-app = DepthAnything3App(
-    model_dir=model_dir,
-    workspace_dir=workspace_dir,
-    gallery_dir=gallery_dir
-)
+MODEL_DIR = os.environ.get("DA3_MODEL_DIR", "depth-anything/DA3NESTED-GIANT-LARGE")
+model = load_model_cpu(MODEL_DIR)
+
+
+# ---------------------------------------------------------
+#  CPU-safe inference (single image only)
+# ---------------------------------------------------------
+
+def run_depth(img):
+    """
+    CPU version of depth inference.
+    No batching, no multiview, no GS, no reconstruction.
+    """
+    if img is None:
+        return None
+
+    with torch.no_grad():
+        depth = model.infer_image(img, device="cpu")
 
-# Optional: pre-cache examples
-examples_dir = os.path.join(workspace_dir, "examples")
-examples_exist = os.path.exists(examples_dir)
-cache_examples_env = os.environ.get("DA3_CACHE_EXAMPLES", "").lower()
-if cache_examples_env in ("false", "0", "no"):
-    cache_examples = False
-elif cache_examples_env in ("true", "1", "yes"):
-    cache_examples = True
-else:
-    cache_examples = examples_exist
-
-cache_gs_tag = os.environ.get("DA3_CACHE_GS_TAG", "dl3dv")
-
-if cache_examples:
-    app.cache_examples(
-        show_cam=True,
-        filter_black_bg=False,
-        filter_white_bg=False,
-        save_percentage=5.0,
-        num_max_points=1000,
-        cache_gs_tag=cache_gs_tag,
-        gs_trj_mode="smooth",
-        gs_video_quality="low",
-    )
+    return depth
+
+
+# ---------------------------------------------------------
+#  Minimal Gradio UI (fast startup)
+# ---------------------------------------------------------
+
+title = "Depth Anything 3 — CPU Mode (Safe HF Version)"
+description = """
+This Hugging Face Space runs **DepthAnything3** in CPU-only mode.
+Only single-image depth inference is enabled.  
+All heavy multiview / GS / reconstruction features were removed so the Space can boot on CPU.
+"""
+
+demo = gr.Interface(
+    fn=run_depth,
+    inputs=gr.Image(type="pil", label="Upload an image"),
+    outputs=gr.Image(label="Predicted Depth"),
+    title=title,
+    description=description,
+)
 
 if __name__ == "__main__":
-    print("🚀 Launching Depth Anything 3 (CPU-only) on Hugging Face Spaces...")
-    app.launch(
-        host="0.0.0.0",
-        port=7860,
-        share=False
-    )
+    demo.launch(server_name="0.0.0.0", server_port=7860)