Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|