zliang commited on
Commit
01666b6
1 Parent(s): 0ff85e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from ultralytics import YOLOv10
10
  def model_cv():
11
  model = YOLOv10("best.pt")
12
 
13
- model.export(format='engine',imgsz=640, iou=0.7, device = 0, simplify=True, half = True, workspace=8)
14
  model = YOLOv10('yolov10l.engine', task='detect')
15
  return model
16
 
 
10
  def model_cv():
11
  model = YOLOv10("best.pt")
12
 
13
+ model.export(format='engine',imgsz=640, iou=0.7, device = 0, simplify=True, half = True)
14
  model = YOLOv10('yolov10l.engine', task='detect')
15
  return model
16