Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -11,8 +11,10 @@ pt_model = YOLO("best.pt")
|
|
11 |
|
12 |
# Export the model to TensorRT format
|
13 |
@spaces.GPU
|
14 |
-
|
|
|
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 |
|