m7mdal7aj commited on
Commit
6cbc475
1 Parent(s): e11d2c6

Update my_model/object_detection.py

Browse files
Files changed (1) hide show
  1. my_model/object_detection.py +1 -0
my_model/object_detection.py CHANGED
@@ -80,6 +80,7 @@ class ObjectDetector:
80
 
81
  try:
82
  model_path = get_model_path ('yolov5')
 
83
  if model_path and os.path.exists(model_path):
84
  self.model = torch.hub.load(model_path, model_version, pretrained=pretrained, source='local')
85
  else:
 
80
 
81
  try:
82
  model_path = get_model_path ('yolov5')
83
+ st.write(model_path)
84
  if model_path and os.path.exists(model_path):
85
  self.model = torch.hub.load(model_path, model_version, pretrained=pretrained, source='local')
86
  else: