Update my_model/object_detection.py

my_model/object_detection.py

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