1

app.py

@@ -47,6 +47,13 @@ silent = False
 gradio_delete_cache = 7200
 print(f'{HERE_PATH}/third_party/ml-depth-pro/checkpoints/')
 hf_hub_download(repo_id="apple/DepthPro", filename='depth_pro.pt', local_dir=f'{HERE_PATH}/third_party/ml-depth-pro/checkpoints/')
+items = os.listdir(f'{HERE_PATH}/third_party/ml-depth-pro/checkpoints/')
+
+# 过滤出文件
+files = [item for item in items if os.path.isfile(os.path.join(directory, item))]
+
+# 打印文件列表
+print(files)
 
 class FileState:
     def __init__(self, outfile_name=None):
@@ -198,7 +205,7 @@ with gradio.Blocks(css=css, title=title, delete_cache=(gradio_delete_cache, grad
             examples=[
                 [
                     os.path.join(HERE_PATH, 'example/bear/00000.jpg'),
-                    2, True, True, True, False, 0.02, "Depth Anything V2",
+                    2, True, True, True, False, 0.02, "Depth Pro",
                      [os.path.join(HERE_PATH, 'example/bear/00000.jpg'),
                       os.path.join(HERE_PATH, 'example/bear/00001.jpg'),
                       os.path.join(HERE_PATH, 'example/bear/00002.jpg'),