Update app.py

app.py

@@ -50,11 +50,11 @@ model = model.to(DEVICE).eval()
 title = "# Depth Anything V2"
 description1 = """Official demo for **Depth Anything V2**.
 Please refer to our [paper](https://arxiv.org/abs/2406.09414) for more details."""
-description2 = """**Due to the issue with our V2 Github repositories, we temporarily upload the content to [Huggingface space](https://huggingface.co/spaces/depth-anything/Depth-Anything-V2/blob/main/README_Github.md)**"""
+description2 = """**Due to the issue with our V2 Github repositories, we temporarily upload the content to [Huggingface space](https://huggingface.co/spaces/depth-anything/Depth-Anything-V2/blob/main/README_Github.md).**"""
 
 @spaces.GPU
 def predict_depth(image):
-    return model.infer_image(image).cpu()
+    return model.infer_image(image)
 
 with gr.Blocks(css=css) as demo:
     gr.Markdown(title)
@@ -78,12 +78,12 @@ with gr.Blocks(css=css) as demo:
 
         depth = predict_depth(image[:, :, ::-1])
 
-        raw_depth = Image.fromarray(depth.numpy().astype('uint16'))
+        raw_depth = Image.fromarray(depth.astype('uint16'))
         tmp_raw_depth = tempfile.NamedTemporaryFile(suffix='.png', delete=False)
         raw_depth.save(tmp_raw_depth.name)
 
         depth = (depth - depth.min()) / (depth.max() - depth.min()) * 255.0
-        depth = depth.numpy().astype(np.uint8)
+        depth = depth.astype(np.uint8)
         colored_depth = (cmap(depth)[:, :, :3] * 255).astype(np.uint8)
 
         gray_depth = Image.fromarray(depth)