inital version

app.py

@@ -18,7 +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
-model = PromptDA.from_pretrained('depth-anything/promptda_vitl').to("cuda").eval()
+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)
 
 def delete_later(path: Union[str, os.PathLike], delay: int = 300):