inital version

app.py

@@ -18,9 +18,10 @@ 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
-import torch
-DEVICE = 'cuda' if torch.cuda.is_available(
-) else 'mps' if torch.backends.mps.is_available() else 'cpu'
+# import torch
+DEVICE = 'cuda' 
+# if torch.cuda.is_available(
+# ) else 'mps' if torch.backends.mps.is_available() else 'cpu'
 model = PromptDA.from_pretrained('depth-anything/promptda_vitl').to(DEVICE).eval()
 thread_pool_executor = ThreadPoolExecutor(max_workers=1)