Update stamp_processing/remover.py

stamp_processing/remover.py

@@ -7,7 +7,7 @@ import numpy.typing as npt
 from .detector import StampDetector
 from .module.unet import *
 from .preprocess import create_batch
-from .utils import REMOVER_WEIGHT_ID, check_image_shape, download_weight, logger
+from .utils import REMOVER_WEIGHT_ID, check_image_shape, download_weight
 
 
 class StampRemover:
@@ -23,16 +23,16 @@ class StampRemover:
                 os.makedirs("tmp/", exist_ok=True)
             removal_weight = os.path.join("tmp", "stamp_remover.pkl")
 
-            logger.info("Downloading stamp remover weight from google drive")
+            print("Downloading stamp remover weight from google drive")
             download_weight(REMOVER_WEIGHT_ID, output=removal_weight)
-            logger.info(f"Finished downloading. Weight is saved at {removal_weight}")
+            print(f"Finished downloading. Weight is saved at {removal_weight}")
 
         try:
             self.remover = UnetInference(removal_weight)  # type: ignore
         except Exception as e:
-            logger.error(e)
-            logger.error("There is something wrong when loading remover weight")
-            logger.error(
+            print(e)
+            print("There is something wrong when loading remover weight")
+            print(
                 "Please make sure you provide the correct path to the weight"
                 "or mannually download the weight at"
                 f"https://drive.google.com/file/d/{REMOVER_WEIGHT_ID}/view?usp=sharing"