Update app.py
Browse files
app.py
CHANGED
|
@@ -48,13 +48,7 @@ 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))]
|
| 55 |
-
|
| 56 |
-
# 打印文件列表
|
| 57 |
-
print(files)
|
| 58 |
|
| 59 |
class FileState:
|
| 60 |
def __init__(self, outfile_name=None):
|
|
@@ -206,7 +200,7 @@ with gradio.Blocks(css=css, title=title, delete_cache=(gradio_delete_cache, grad
|
|
| 206 |
examples=[
|
| 207 |
[
|
| 208 |
os.path.join(HERE_PATH, 'example/bear/00000.jpg'),
|
| 209 |
-
2, True, True, True, False, 0.02, "Depth
|
| 210 |
[os.path.join(HERE_PATH, 'example/bear/00000.jpg'),
|
| 211 |
os.path.join(HERE_PATH, 'example/bear/00001.jpg'),
|
| 212 |
os.path.join(HERE_PATH, 'example/bear/00002.jpg'),
|
|
|
|
| 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 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
|
| 53 |
class FileState:
|
| 54 |
def __init__(self, outfile_name=None):
|
|
|
|
| 200 |
examples=[
|
| 201 |
[
|
| 202 |
os.path.join(HERE_PATH, 'example/bear/00000.jpg'),
|
| 203 |
+
2, True, True, True, False, 0.02, "Depth Anything V2",
|
| 204 |
[os.path.join(HERE_PATH, 'example/bear/00000.jpg'),
|
| 205 |
os.path.join(HERE_PATH, 'example/bear/00001.jpg'),
|
| 206 |
os.path.join(HERE_PATH, 'example/bear/00002.jpg'),
|