Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ INPUT_WIDTH = 640
|
|
31 |
INPUT_HEIGHT = 640
|
32 |
|
33 |
# load YOLO model
|
34 |
-
net = cv2.dnn.readNetFromONNX('
|
35 |
net.setPreferableBackend(cv2.dnn.DNN_BACKEND_OPENCV)
|
36 |
net.setPreferableTarget(cv2.dnn.DNN_TARGET_CPU)
|
37 |
|
|
|
31 |
INPUT_HEIGHT = 640
|
32 |
|
33 |
# load YOLO model
|
34 |
+
net = cv2.dnn.readNetFromONNX('./best.onnx')
|
35 |
net.setPreferableBackend(cv2.dnn.DNN_BACKEND_OPENCV)
|
36 |
net.setPreferableTarget(cv2.dnn.DNN_TARGET_CPU)
|
37 |
|