Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -189,7 +189,7 @@ import torch
|
|
| 189 |
import cv2
|
| 190 |
|
| 191 |
# Load the YOLOv8 model
|
| 192 |
-
model = torch.hub.load('ultralytics/yolov8', 'yolov8s')
|
| 193 |
model.load_state_dict(torch.load('Model_IV'))
|
| 194 |
|
| 195 |
def inference(img):
|
|
|
|
| 189 |
import cv2
|
| 190 |
|
| 191 |
# Load the YOLOv8 model
|
| 192 |
+
model = torch.hub.load('ultralytics/yolov8', 'yolov8s', trust_repo=True)
|
| 193 |
model.load_state_dict(torch.load('Model_IV'))
|
| 194 |
|
| 195 |
def inference(img):
|