v05 update

app.py

@@ -48,7 +48,7 @@ ROOT_PATH = sys.path[0]  # 根目录
 yolov5_path = "ultralytics/yolov5"
 
 # 本地模型路径
-local_model_path = f"{ROOT_PATH}/models"
+# local_model_path = f"{ROOT_PATH}/models"
 
 # Gradio YOLOv5 Det版本
 GYD_VERSION = "Gradio YOLOv5 Det v0.5"
@@ -161,8 +161,7 @@ def model_loading(model_name, device, opt=[]):
         torch.hub._validate_not_a_forked_repo = lambda a, b, c: True
         model = torch.hub.load(
             yolov5_path,
-            "custom",
-            path=f"{local_model_path}/{model_name}",
+            path=model_name,
             device=device,
             force_reload=[True if "refresh_yolov5" in opt else False][0],
             _verbose=True,
@@ -661,7 +660,7 @@ def main(args):
     inputs_maxnum01 = gr.Number(value=max_detnum, label="最大检测数")
     inputs_clsName01 = gr.CheckboxGroup(choices=model_cls_name, value=model_cls_name, type="index", label="类别")
     inputs_opt01 = gr.CheckboxGroup(choices=["refresh_yolov5", "label", "pdf", "json", "csv", "excel"],
-                                    value=["refresh_yolov5", "label", "pdf"],
+                                    value=["label", "pdf"],
                                     type="value",
                                     label="操作")