Update engine.py

engine.py

@@ -16,15 +16,15 @@ from .stamp_processing.callback import remove_potiential_table_fp, remove_box_by
 class Engine(f.Pipe):
 
     def __init__(self, cfg:Config=None, table_checkpoint=None, line_checkpoint=None, verbose=True):
-        super().__init__()
-        self.cfg = cfg or Config()
-        self.standard_width = cfg.STANDARD_WIDTH
-        self.preprocessor = Preprocessor(cfg.STANDARD_WIDTH, cfg.STANDARD_INTERPOLATION)
-        self.post_processor = TableRestore()
-        self.table_detector = self._build_table_detector(cfg, table_checkpoint)
-        self.stamp_detector = StampDetector(model_path=cfg.STAMP_DETECTION_CHECKPOINT, device=cfg.DEVICE)
+        super().__init__(
+            cfg = cfg or Config(),
+            standard_width = cfg.STANDARD_WIDTH,
+            preprocessor = Preprocessor(cfg.STANDARD_WIDTH, cfg.STANDARD_INTERPOLATION),
+            post_processor = TableRestore(),
+            table_detector = self._build_table_detector(cfg, table_checkpoint),
+            stamp_detector = StampDetector(model_path=cfg.STAMP_DETECTION_CHECKPOINT, device=cfg.DEVICE)
+        )
         self.stamp_detector.model.to(cfg.DEVICE)
-
     def run(self, image):
         # ratio = 1.0*self.standard_width/image.shape[1]
         # image = self.preprocessor(image)