Spaces:

Sa-m
/

YoloV5-Party-Symbol-Detector-V1

Runtime error

Sa-m commited on Feb 18, 2022

Commit

1805721

•

1 Parent(s): 814d854

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ def load_model():
  '''
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/best.pt')
  model.conf = 0.38
- model.hide_conf=True
  model.dnn=True
  model.line_thickness=6
  return model

  '''
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/best.pt')
  model.conf = 0.38
+ model.hide_labels=True
  model.dnn=True
  model.line_thickness=6
  return model