tori29umai commited on
Commit
4cc03e0
·
verified ·
1 Parent(s): b079214

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -7,6 +7,7 @@ from PIL import Image
7
  import trimesh
8
  from huggingface_hub import hf_hub_download
9
  from depth_anything_v2.dpt import DepthAnythingV2
 
10
 
11
  css = """
12
  #img-display-container {
@@ -53,6 +54,7 @@ description = """
53
  - [プロジェクトページ](https://depth-anything-v2.github.io)
54
  - [GitHub](https://github.com/DepthAnything/Depth-Anything-V2)
55
  """
 
56
  def predict_depth(image):
57
  return model.infer_image(image)
58
 
 
7
  import trimesh
8
  from huggingface_hub import hf_hub_download
9
  from depth_anything_v2.dpt import DepthAnythingV2
10
+ import spaces
11
 
12
  css = """
13
  #img-display-container {
 
54
  - [プロジェクトページ](https://depth-anything-v2.github.io)
55
  - [GitHub](https://github.com/DepthAnything/Depth-Anything-V2)
56
  """
57
+ @spaces.GPU
58
  def predict_depth(image):
59
  return model.infer_image(image)
60