Spaces:

Sa-m
/

YoloV5-Party-Symbol-Detector-V1

Runtime error

Sa-m commited on Feb 18, 2022

Commit

3a05e49

•

1 Parent(s): e35b449

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

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