Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -72,9 +72,6 @@ class ModelManager:
|
|
72 |
@property
|
73 |
def yolo_model(self):
|
74 |
if self._yolo_model is None:
|
75 |
-
import torch.serialization
|
76 |
-
from ultralytics.nn.tasks import DetectionModel
|
77 |
-
torch.serialization.add_safe_globals([DetectionModel])
|
78 |
self._yolo_model = YOLO('yolov8x.pt')
|
79 |
return self._yolo_model
|
80 |
|
|
|
72 |
@property
|
73 |
def yolo_model(self):
|
74 |
if self._yolo_model is None:
|
|
|
|
|
|
|
75 |
self._yolo_model = YOLO('yolov8x.pt')
|
76 |
return self._yolo_model
|
77 |
|