Update stamp_processing/detector.py

stamp_processing/detector.py

@@ -24,7 +24,7 @@ class StampDetector:
     ) -> None:
         """Create an object for stamp detection"""
         # assert device == "cpu", "Currently only support cpu inference"
-
+        
         checkpoint = hf.download_file(model_path)
         print(model_path)
         print(checkpoint)
@@ -36,7 +36,7 @@ class StampDetector:
         self.conf_thres = conf_thres
         self.iou_thres = iou_thres
 
-        self.process_func_ = partial(process_image, device=self.device)
+        self.process_func_ = partial(process_image, device="cpu")
 
     def __call__(self, image_list: Union[List[npt.NDArray], npt.NDArray]) -> List[npt.NDArray]:
         """Returns a list of bounding boxes [xmin, ymin, xmax, ymax] for each image in image_list