inital version

app.py

@@ -15,8 +15,8 @@ import cv2
 import numpy as np
 import imageio
 # from promptda.promptda import PromptDA
-from promptda.utils.io_wrapper import load_image, load_depth
-from promptda.utils.depth_utils import visualize_depth, unproject_depth
+# from promptda.utils.io_wrapper import load_image, load_depth
+# from promptda.utils.depth_utils import visualize_depth, unproject_depth
 # import torch
 DEVICE = 'cuda' 
 # if torch.cuda.is_available(
@@ -44,7 +44,7 @@ def delete_later(path: Union[str, os.PathLike], delay: int = 300):
     atexit.register(_delete)
 
 
-@spaces.GPU
+# @spaces.GPU
 def run_with_gpu(image, prompt_depth):
     image = image.to(DEVICE)
     prompt_depth = prompt_depth.to(DEVICE)
@@ -57,6 +57,7 @@ def check_is_stray_scanner_app_capture(input_dir):
     assert os.path.exists(os.path.join(input_dir, 'rgb.mp4')), 'rgb.mp4 not found'
     pass
 
+@spaces.GPU
 def run(input_file, resolution):
     # unzip zip file
     input_file = input_file.name