app.py
CHANGED
|
@@ -47,7 +47,8 @@ silent = False
|
|
| 47 |
gradio_delete_cache = 7200
|
| 48 |
print(f'{HERE_PATH}/third_party/ml-depth-pro/checkpoints/')
|
| 49 |
hf_hub_download(repo_id="apple/DepthPro", filename='depth_pro.pt', local_dir=f'{HERE_PATH}/third_party/ml-depth-pro/checkpoints/')
|
| 50 |
-
|
|
|
|
| 51 |
|
| 52 |
# 过滤出文件
|
| 53 |
files = [item for item in items if os.path.isfile(os.path.join(directory, item))]
|
|
|
|
| 47 |
gradio_delete_cache = 7200
|
| 48 |
print(f'{HERE_PATH}/third_party/ml-depth-pro/checkpoints/')
|
| 49 |
hf_hub_download(repo_id="apple/DepthPro", filename='depth_pro.pt', local_dir=f'{HERE_PATH}/third_party/ml-depth-pro/checkpoints/')
|
| 50 |
+
directory = f'{HERE_PATH}/third_party/ml-depth-pro/checkpoints/'
|
| 51 |
+
items = os.listdir(directory)
|
| 52 |
|
| 53 |
# 过滤出文件
|
| 54 |
files = [item for item in items if os.path.isfile(os.path.join(directory, item))]
|