zliang commited on
Commit
e8f8554
·
verified ·
1 Parent(s): ef5c77e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -10,7 +10,12 @@ from ultralytics import YOLO
10
  @spaces.GPU
11
  def model_cov():
12
  model_pt = YOLO("best.pt")
13
- model_pt.export(format="engine") # creates 'yolov8n.engine'
 
 
 
 
 
14
  model =YOLO("best.engine")
15
 
16
  return model
 
10
  @spaces.GPU
11
  def model_cov():
12
  model_pt = YOLO("best.pt")
13
+ model.export(
14
+ format="engine",
15
+ dynamic=True,
16
+ batch=8,
17
+ workspace=4,
18
+ )
19
  model =YOLO("best.engine")
20
 
21
  return model