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