DepthPro-transformers-Grayscale

Sleeping

hysts HF Staff commited on Feb 18

Commit

7a98e98

1 Parent(s): b1ed4fe

Update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ model = DepthProForDepthEstimation.from_pretrained("apple/DepthPro-hf").to(devic
 cmap = mpl.colormaps.get_cmap("Spectral_r")
-@spaces.GPU(duration=10)
 @torch.inference_mode()
 def run(image: PIL.Image.Image) -> tuple[tuple[PIL.Image.Image, PIL.Image.Image], str, str, str, str]:
     inputs = image_processor(images=image, return_tensors="pt").to(device)

 cmap = mpl.colormaps.get_cmap("Spectral_r")
+@spaces.GPU(duration=20)
 @torch.inference_mode()
 def run(image: PIL.Image.Image) -> tuple[tuple[PIL.Image.Image, PIL.Image.Image], str, str, str, str]:
     inputs = image_processor(images=image, return_tensors="pt").to(device)