feat: donwload progress bar

README.md

@@ -50,7 +50,7 @@
 
 - 在线体验： [![SwanHub Demo](https://img.shields.io/static/v1?label=Demo&message=SwanHub%20Demo&color=blue)](https://swanhub.co/ZeYiLin/HivisionIDPhotos/demo)、[![Spaces](https://img.shields.io/badge/🤗-Open%20in%20Spaces-blue)](https://huggingface.co/spaces/TheEeeeLin/HivisionIDPhotos)
 
-- 2024.09.08: 增加新的抠图模型 [RMBG-1.4](https://huggingface.co/briaai/RMBG-1.4) | ComfyUI工作流 - [HivisionIDPhotos-ComfyUI](https://github.com/AIFSH/HivisionIDPhotos-ComfyUI) 贡献 by [AIFSH](https://github.com/AIFSH/HivisionIDPhotos-ComfyUI)
+- 2024.09.08: 增加新的**抠图模型** [RMBG-1.4](https://huggingface.co/briaai/RMBG-1.4) | **ComfyUI工作流** - [HivisionIDPhotos-ComfyUI](https://github.com/AIFSH/HivisionIDPhotos-ComfyUI) 贡献 by [AIFSH](https://github.com/AIFSH/HivisionIDPhotos-ComfyUI)
 - 2024.09.07: 增加**人脸检测API选项** [Face++](docs/face++_CN.md)，实现更高精度的人脸检测
 - 2024.09.06: 增加新的抠图模型 [modnet_photographic_portrait_matting.onnx](https://github.com/ZHKKKe/MODNet)
 - 2024.09.05: 更新 [Restful API 文档](docs/api_CN.md)

scripts/download_model.py

@@ -1,6 +1,7 @@
 import os
 import requests
 import argparse
+from tqdm import tqdm  # 导入 tqdm 库
 
 # 获取当前脚本所在目录的上一级目录
 base_path = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
@@ -11,9 +12,20 @@ def download_file(url, save_path):
         print(f"Begin downloading: {url}")
         response = requests.get(url, stream=True)
         response.raise_for_status()  # 检查请求是否成功
-        with open(save_path, "wb") as file:
+
+        # 获取文件总大小
+        total_size = int(response.headers.get("content-length", 0))
+        # 使用 tqdm 显示进度条
+        with open(save_path, "wb") as file, tqdm(
+            total=total_size,
+            unit="B",
+            unit_scale=True,
+            unit_divisor=1024,
+            desc=os.path.basename(save_path),
+        ) as bar:
             for chunk in response.iter_content(chunk_size=8192):
                 file.write(chunk)
+                bar.update(len(chunk))  # 更新进度条
         print(f"Download completed. Save to: {save_path}")
     except requests.exceptions.RequestException as e:
         print(f"Download failed: {e}")