fix yaml load

app.py

@@ -10,12 +10,11 @@ import gradio as gr
 import torch
 import yaml
 from PIL import Image
-from zmq import device
 
 ROOT_PATH = sys.path[0]  # 根目录
 
-# 本地模型路径
-local_model_path = f"{ROOT_PATH}/yolov5"
+# 模型路径
+model_path = "ultralytics/yolov5"
 
 
 # 模型名称临时变量
@@ -67,7 +66,7 @@ def parse_args(known=False):
         "-dev",
         default="cpu",
         type=str,
-        help="cuda or cpu",
+        help="cuda or cpu, hugging face only cpu",
     )
 
     args = parser.parse_known_args()[0] if known else parser.parse_args()
@@ -78,18 +77,7 @@ def parse_args(known=False):
 def model_loading(model_name, device):
 
     # 加载本地模型
-    # model = torch.hub.load('ultralytics/yolov5', 'yolov5s', device=device)
-    model = torch.hub.load(
-        "ultralytics/yolov5", model_name, force_reload=True, device=device
-    )
-
-    # model = torch.hub.load(
-    #     local_model_path,
-    #     "custom",
-    #     path=f"{local_model_path}/{model_name}",
-    #     source="local",
-    #     device=device,
-    # )
+    model = torch.hub.load(model_path, model_name, force_reload=True, device=device)
 
     return model
 
@@ -151,9 +139,7 @@ def yolo_det(img, device, model_name, conf, iou, label_opt, model_cls):
 
 # yaml文件解析
 def yaml_parse(file_path):
-    return yaml.load(
-        open(file_path, "r", encoding="utf-8").read(), Loader=yaml.FullLoader
-    )
+    return yaml.safe_load(open(file_path, "r", encoding="utf-8").read())
 
 
 def main(args):
@@ -182,7 +168,7 @@ def main(args):
     # -------------------输入组件-------------------
     inputs_img = gr.inputs.Image(type="pil", label="原始图片")
     device = gr.inputs.Dropdown(
-        choices=["0", "cpu"], default=device, type="value", label="设备"
+        choices=["cpu"], default=device, type="value", label="设备"
     )
     inputs_model = gr.inputs.Dropdown(
         choices=model_names, default=model_name, type="value", label="模型"