zliang commited on
Commit
326638c
·
verified ·
1 Parent(s): f91708f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -14
app.py CHANGED
@@ -7,20 +7,8 @@ from ultralytics import YOLO
7
  # Load the trained model
8
 
9
 
10
- @spaces.GPU
11
- def model_cov():
12
- model_pt = YOLO("best.pt")
13
- model_pt.export(
14
- format="engine",
15
- dynamic=True,
16
- batch=8,
17
- workspace=4,
18
- )
19
- model =YOLO("best.engine")
20
-
21
- return model
22
-
23
- model =model_cov()
24
 
25
 
26
 
 
7
  # Load the trained model
8
 
9
 
10
+ model_pt = YOLO("best.pt")
11
+
 
 
 
 
 
 
 
 
 
 
 
 
12
 
13
 
14