zliang commited on
Commit
7aab8c4
·
verified ·
1 Parent(s): 337f0b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -6
app.py CHANGED
@@ -7,13 +7,8 @@ import spaces
7
  # Load the trained model
8
 
9
 
10
- @spaces.GPU
11
- def model_c():
12
- model = YOLO("best.pt")
13
-
14
 
15
- model=model_c()
16
- # Load the exported TensorRT model
17
 
18
 
19
  # Define the class indices for figures and tables
 
7
  # Load the trained model
8
 
9
 
 
 
 
 
10
 
11
+ model = YOLO("best.engine")
 
12
 
13
 
14
  # Define the class indices for figures and tables