Update my_model/object_detection.py

my_model/object_detection.py

@@ -6,6 +6,7 @@ import numpy as np
 import cv2
 import os
 from my_model.utilities import get_path, show_image
+import models.deformable-detr-detic as detic
 
 
 
@@ -61,7 +62,8 @@ class ObjectDetector:
         """
 
         try:
-            model_path = get_path('deformable-detr-detic', 'models')
+            #model_path = get_path('deformable-detr-detic', 'models')  # for use locally
+            model_path = detic  # for use on HF only
             self.processor = AutoImageProcessor.from_pretrained(model_path)
             self.model = AutoModelForObjectDetection.from_pretrained(model_path)
         except Exception as e: