add model auto downloads

app.py

@@ -27,11 +27,11 @@ data_url_dict = {
 }
 
 model_url_dict = {
-    "cnn_se.pt": "https://huggingface.co/Tsumugii/lesion-cells-det/raw/main/cnn_se.pt",
-    "detr_based.pt": "https://huggingface.co/Tsumugii/lesion-cells-det/raw/main/detr_based.pt",
-    "vit_based.pt": "https://huggingface.co/Tsumugii/lesion-cells-det/raw/main/vit_based.pt",
-    "yolov5_based.pt": "https://huggingface.co/Tsumugii/lesion-cells-det/raw/main/yolov5_based.pt",
-    "yolov8_based.pt": "https://huggingface.co/Tsumugii/lesion-cells-det/raw/main/yolov8_based.pt",
+    "cnn_se.pt": "https://huggingface.co/Tsumugii/lesion-cells-det/resolve/main/cnn_se.pt",
+    "detr_based.pt": "https://huggingface.co/Tsumugii/lesion-cells-det/resolve/main/detr_based.pt",
+    "vit_based.pt": "https://huggingface.co/Tsumugii/lesion-cells-det/resolve/main/vit_based.pt",
+    "yolov5_based.pt": "https://huggingface.co/Tsumugii/lesion-cells-det/resolve/main/yolov5_based.pt",
+    "yolov8_based.pt": "https://huggingface.co/Tsumugii/lesion-cells-det/resolve/main/yolov8_based.pt",
 }
 
 # 判断字体文件是否存在
@@ -77,7 +77,7 @@ def download_fonts(font_diff):
     for k, v in data_url_dict.items():
         if k in font_diff:
             font_name = v.split("/")[-1]  # 字体名称
-            fonts_directory_path.mkdir(parents=True, exist_ok=True)  # 创建目录
+            fonts_directory_path.mkdir(parents=True, exist_ok=True)  # 创建本地字体目录
 
             font_file_path = f"{ROOT_PATH}/fonts/{font_name}"  # 字体路径
             # 下载字体文件
@@ -87,13 +87,14 @@ def download_fonts(font_diff):
 def download_models(model_diff):
     global model_name
 
-    for k in model_diff:
-        v = model_url_dict[k]
-        model_name = v.split("/")[-1]  # 模型名称
+    for k, v in model_url_dict.items():
+        if k in model_diff:
+            model_name = v.split("/")[-1]  # 模型名称
+            models_directory_path.mkdir(parents=True, exist_ok=True)  # 创建本地模型目录
 
-        model_file_path = f"{ROOT_PATH}/models/{model_name}"  # 模型路径
-        # 下载模型文件
-        wget.download(v, model_file_path)
+            model_file_path = f"{ROOT_PATH}/models/{model_name}"  # 模型路径
+            # 下载模型文件
+            wget.download(v, model_file_path)
 
 
 is_fonts(fonts_directory_path)