Update app.py

app.py

@@ -11,6 +11,7 @@ from torch import Tensor
 from genstereo import GenStereo, AdaptiveFusionLayer
 import ssl
 from huggingface_hub import hf_hub_download
+import spaces
 
 from extern.DAM2.depth_anything_v2.dpt import DepthAnythingV2
 ssl._create_default_https_context = ssl._create_unverified_context
@@ -178,6 +179,7 @@ with tempfile.TemporaryDirectory() as tmpdir:
             return (disp - disp.min()) / (disp.max() - disp.min())
 
         # Callbacks
+        @spaces.GPU()        
         def cb_mde(image_file: str):
             if not image_file:
                 # Return None if no image is provided (e.g., when file is cleared).
@@ -195,7 +197,7 @@ with tempfile.TemporaryDirectory() as tmpdir:
 
             return image, depth_image, image, depth
 
-
+        @spaces.GPU()
         def cb_generate(image, depth: Tensor, scale_factor):
             norm_disp = normalize_disp(depth)
             disp = norm_disp * scale_factor / 100 * IMAGE_SIZE