zliang commited on
Commit
0014b69
·
verified ·
1 Parent(s): 819efd7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -11,8 +11,10 @@ pt_model = YOLO("best.pt")
11
 
12
  # Export the model to TensorRT format
13
  @spaces.GPU
14
- pt_model.export(format="engine") # creates 'yolov8n.engine'
 
15
 
 
16
  # Load the exported TensorRT model
17
  model = YOLO("best.engine")
18
 
 
11
 
12
  # Export the model to TensorRT format
13
  @spaces.GPU
14
+ def model_c():
15
+ pt_model.export(format="engine") # creates 'yolov8n.engine'
16
 
17
+ model_c()
18
  # Load the exported TensorRT model
19
  model = YOLO("best.engine")
20