Update app.py

app.py

@@ -117,7 +117,7 @@ def export_json(results, model, img_size):
     ]
 
 
-def yolo_det(img, experiment_id, device, model_name, inference_size, conf, iou, label_opt, model_cls):
+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
 
@@ -227,13 +227,13 @@ def main(args):
     inputs = [
         inputs_img,  # 输入图片
         experiment_id, # path adding scheduler
-        device,  # 设备
-        inputs_model,  # 模型
-        inputs_size,  # 推理尺寸
-        input_conf,  # 置信度阈值
-        inputs_iou,  # IoU阈值
-        inputs_label,  # 标签显示
-        inputs_clsName,  # 类别
+        # device,  # 设备
+        # inputs_model,  # 模型
+        # inputs_size,  # 推理尺寸
+        # input_conf,  # 置信度阈值
+        # inputs_iou,  # IoU阈值
+        # inputs_label,  # 标签显示
+        # inputs_clsName,  # 类别
     ]
     # 输出参数
     outputs = gr.outputs.Image(type="pil", label="检测图片")