m7mdal7aj commited on
Commit
ae755df
·
verified ·
1 Parent(s): 58b59bc

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
@@ -61,6 +61,7 @@ class ObjectDetector:
61
 
62
  try:
63
  model_path = get_model_path('deformable-detr-detic')
 
64
  self.processor = AutoImageProcessor.from_pretrained(model_path)
65
  self.model = AutoModelForObjectDetection.from_pretrained(model_path)
66
  except Exception as e:
 
61
 
62
  try:
63
  model_path = get_model_path('deformable-detr-detic')
64
+ st.write(model_path)
65
  self.processor = AutoImageProcessor.from_pretrained(model_path)
66
  self.model = AutoModelForObjectDetection.from_pretrained(model_path)
67
  except Exception as e: