MedCat
/

table_detection_3

nguyenp99 commited on Apr 23

Commit

1c6f7e9

•

1 Parent(s): 5ae5318

Update stamp_processing/remover.py

Files changed (1) hide show

stamp_processing/remover.py CHANGED Viewed

@@ -39,7 +39,7 @@ class StampRemover:
             )
             raise FileNotFoundError()
-        self.detector = StampDetector(detection_weight, device=device)
         self.padding = 3
     def __call__(self, image_list: Union[List[npt.NDArray], npt.NDArray], batch_size: int = 16) -> List[npt.NDArray]:

             )
             raise FileNotFoundError()
+        self.detector = StampDetector(detection_weight, device="cpu")
         self.padding = 3
     def __call__(self, image_list: Union[List[npt.NDArray], npt.NDArray], batch_size: int = 16) -> List[npt.NDArray]: