Update app.py

app.py

@@ -1,9 +1,3 @@
-# Gradio YOLOv5 Det v0.1
-# 创建人：曾逸夫
-# 创建时间：2022-04-03
-# email：[email protected]
-# 项目主页：https://gitee.com/CV_Lab/gradio_yolov5_det
-
 import argparse
 import csv
 import sys
@@ -18,20 +12,14 @@ ROOT_PATH = sys.path[0]  # 根目录
 
 # 模型路径
 model_path = "ultralytics/yolov5"
-
-
 # 模型名称临时变量
 model_name_tmp = ""
-
 # 设备临时变量
 device_tmp = ""
-
 # 文件后缀
 suffix_list = [".csv", ".yaml"]
-
-
 def parse_args(known=False):
-    parser = argparse.ArgumentParser(description="Gradio YOLOv5 Det v0.1")
+    parser = argparse.ArgumentParser(description="Gradio LIVE")
     parser.add_argument(
         "--model_name", "-mn", default="yolov5s", type=str, help="model name"
     )
@@ -80,8 +68,6 @@ def parse_args(known=False):
 
     args = parser.parse_known_args()[0] if known else parser.parse_args()
     return args
-
-
 #  模型加载
 def model_loading(model_name, device):
 
@@ -89,8 +75,6 @@ def model_loading(model_name, device):
     model = torch.hub.load(model_path, model_name, force_reload=True, device=device)
 
     return model
-
-
 # 检测信息
 def export_json(results, model, img_size):
 
@@ -115,8 +99,6 @@ def export_json(results, model, img_size):
         ]
         for result in results.xyxyn
     ]
-
-
 def yolo_det(img, experiment_id, device=None, model_name=None, inference_size=None, conf=None, iou=None, label_opt=None, model_cls=None):
 
     global model, model_name_tmp, device_tmp
@@ -145,6 +127,13 @@ def yolo_det(img, experiment_id, device=None, model_name=None, inference_size=No
     return det_img, det_json
 
 
+
+
+
+
+
+
+
 # yaml文件解析
 def yaml_parse(file_path):
     return yaml.safe_load(open(file_path, "r", encoding="utf-8").read())
@@ -193,8 +182,8 @@ def main(args):
     inputs_img = gr.inputs.Image(type="pil", label="Input Image")
     experiment_id = gr.inputs.Radio(
         choices=[
-            "add [1,1,1,1,1] total 5 paths",
-            "add [1,1,1,1,1,1,1,1] total 8 paths",
+            "add [1, 1, 1, 1, 1] total 5 paths",
+            "add [1, 1, 1, 1, 1, 1, 1, 1] total 8 paths",
             "add [1,2,4,8,16,32, ...] total 128 paths",
             "add [1,2,4,8,16,32, ...] total 256 paths"], type="value", default="add [1,1,1,1,1] paths", label="Path Adding Scheduler"
     )
@@ -248,11 +237,11 @@ def main(args):
     examples = [
         [
             "./examples/1.png",
-            "add [1,1,1,1,1] total 5 paths",
+            "add [1, 1, 1, 1, 1] total 5 paths",
         ],
         [
             "./examples/2.png",
-            "add [1,1,1,1,1] total 5 paths",
+            "add [1, 1, 1, 1, 1] total 5 paths",
         ],
         [
             "./examples/3.jpg",
@@ -264,7 +253,7 @@ def main(args):
         ],
         [
             "./examples/5.png",
-            "add [1,1,1,1,1] total 5 paths",
+            "add [1, 1, 1, 1, 1] total 5 paths",
         ],
     ]
 
@@ -274,7 +263,7 @@ def main(args):
         inputs=inputs,
         outputs=[outputs, outputs02],
         title=title,
-        # description=description,
+        description=description,
         examples=examples,
         theme="seafoam",
         # live=True, # 实时变更输出