v02 update

app.py

@@ -20,8 +20,8 @@ from util.pdf_opt import pdf_generate
 
 ROOT_PATH = sys.path[0]  # 根目录
 
-# 本地模型路径
-local_model_path = f"{ROOT_PATH}/yolov5"
+# 模型路径
+model_path = "ultralytics/yolov5"
 
 # 模型名称临时变量
 model_name_tmp = ""
@@ -112,14 +112,9 @@ def yaml_csv(file_path, file_tag):
 #  模型加载
 def model_loading(model_name, device):
 
-    # 加载本地模型
+    # 加载模型
     model = torch.hub.load(
-        local_model_path,
-        "custom",
-        path=f"{local_model_path}/{model_name}",
-        source="local",
-        device=device,
-        _verbose=False,
+        model_path, model_name, force_reload=True, device=device, _verbose=False
     )
 
     return model