zliang commited on
Commit
0ff85e0
1 Parent(s): 77605d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from ultralytics import YOLOv10
7
  # Load the trained model
8
 
9
  @spaces.GPU
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)
 
7
  # Load the trained model
8
 
9
  @spaces.GPU
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)