Update my_model/object_detection.py

my_model/object_detection.py

@@ -66,7 +66,7 @@ class ObjectDetector:
         try:
             model_path = get_model_path('deformable-detr-detic')
             self.processor = AutoImageProcessor.from_pretrained(model_path)
-            self.model = AutoModelForObjectDetection.from_pretrained(model_path, device_map = self.device)
+            self.model = AutoModelForObjectDetection.from_pretrained(model_path)
         except Exception as e:
             print(f"Error loading Detic model: {e}")
             raise