zliang commited on
Commit
819efd7
·
verified ·
1 Parent(s): 19e0bca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -10,6 +10,7 @@ import spaces
10
  pt_model = YOLO("best.pt")
11
 
12
  # Export the model to TensorRT format
 
13
  pt_model.export(format="engine") # creates 'yolov8n.engine'
14
 
15
  # Load the exported TensorRT model
 
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