tori29umai
commited on
Update app.py
Browse files
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 |
|