DawnC commited on
Commit
7b8cadb
·
verified ·
1 Parent(s): 88ad472

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
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