Update app.py

app.py

@@ -4,7 +4,6 @@ import matplotlib
 import numpy as np
 import os
 from PIL import Image
-import spaces
 import torch
 import torch.nn.functional as F
 import open3d as o3d
@@ -18,6 +17,12 @@ from ppd.utils.depth2pcd import depth2pcd
 from moge.model.v2 import MoGeModel 
 from ppd.models.ppd import PixelPerfectDepth
 
+try:
+    import spaces
+    HUGGINFACE_SPACES_INSTALLED = True
+except ImportError:
+    HUGGINFACE_SPACES_INSTALLED = False
+
 css = """
 #img-display-container {
     max-height: 100vh;
@@ -66,16 +71,18 @@ def main(share=True):
     description = """Official demo for **Pixel-Perfect Depth**.
     Please refer to our [paper](), [project page](https://pixel-perfect-depth.github.io), and [github](https://github.com/gangweix/pixel-perfect-depth) for more details."""
 
-    @spaces.GPU
+    @(spaces.GPU if HUGGINFACE_SPACES_INSTALLED else (lambda x: x))
     def predict_depth(image, denoise_steps):
+        global model
         model = model.to(DEVICE)
         depth, resize_image = model.infer_image(image, sampling_steps=denoise_steps)
         return depth, resize_image
 
-    @spaces.GPU
+    @(spaces.GPU if HUGGINFACE_SPACES_INSTALLED else (lambda x: x))
     def predict_moge_depth(image):
         image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
         image = torch.tensor(image / 255, dtype=torch.float32, device=DEVICE).permute(2, 0, 1)
+        global moge_model
         moge_model = moge_model.to(DEVICE)
         metric_depth, mask, intrinsics = moge_model.infer(image)
         metric_depth[~mask] = metric_depth[mask].max()