Update app.py

app.py

@@ -146,6 +146,7 @@ with tempfile.TemporaryDirectory() as tmpdir:
                 print(f"Switched to GenStereo {sd_version_choice}. IMAGE_SIZE: {IMAGE_SIZE}, CHECKPOINT: {CHECKPOINT_NAME}")
             return None, None, None, None, None, None
 
+        @spaces.GPU()
         def cb_mde(image_file: str, sd_version):
             if not image_file:
                 # Return None if no image is provided (e.g., when file is cleared).
@@ -171,6 +172,7 @@ with tempfile.TemporaryDirectory() as tmpdir:
 
             return image, depth_image, image, depth
 
+        @spaces.GPU()
         def cb_generate(image, depth: Tensor, scale_factor, sd_version):
             norm_disp = normalize_disp(depth.cuda())
             disp = norm_disp * scale_factor / 100 * IMAGE_SIZE