app.py
CHANGED
|
@@ -28,6 +28,7 @@ from transformers import pipeline
|
|
| 28 |
from dust3r.utils.viz_demo import convert_scene_output_to_glb
|
| 29 |
import depth_pro
|
| 30 |
import spaces
|
|
|
|
| 31 |
pl.ion()
|
| 32 |
|
| 33 |
# for gpu >= Ampere and pytorch >= 1.12
|
|
@@ -39,6 +40,7 @@ image_size = 512
|
|
| 39 |
silent = True
|
| 40 |
gradio_delete_cache = 7200
|
| 41 |
|
|
|
|
| 42 |
|
| 43 |
class FileState:
|
| 44 |
def __init__(self, outfile_name=None):
|
|
|
|
| 28 |
from dust3r.utils.viz_demo import convert_scene_output_to_glb
|
| 29 |
import depth_pro
|
| 30 |
import spaces
|
| 31 |
+
from huggingface_hub import hf_hub_download
|
| 32 |
pl.ion()
|
| 33 |
|
| 34 |
# for gpu >= Ampere and pytorch >= 1.12
|
|
|
|
| 40 |
silent = True
|
| 41 |
gradio_delete_cache = 7200
|
| 42 |
|
| 43 |
+
hf_hub_download(repo_id="apple/DepthPro", filename='depth_pro.pt', local_dir='third_party/ml-depth-pro/checkpoints/')
|
| 44 |
|
| 45 |
class FileState:
|
| 46 |
def __init__(self, outfile_name=None):
|