atalaydenknalbant commited on
Commit
ad343a5
·
verified ·
1 Parent(s): 0b7100f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def yolo_inference(input_type, image, video, model_id, conf_threshold, iou_thres
16
  Accepts detect/seg/pose/obb/cls checkpoints and renders r.plot().
17
  """
18
  model = YOLO(model_id)
19
-
20
  if input_type == "Image":
21
  if image is None:
22
  w, h = 640, 480
 
16
  Accepts detect/seg/pose/obb/cls checkpoints and renders r.plot().
17
  """
18
  model = YOLO(model_id)
19
+ del model.model.model[-1].one2one_cv2
20
  if input_type == "Image":
21
  if image is None:
22
  w, h = 640, 480