Update my_model/object_detection.py

my_model/object_detection.py

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