Update engine.py

engine.py

@@ -23,7 +23,7 @@ class Engine(f.Pipe):
         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)
-        self.stamp_detector.to(cfg.DEVICE)
+        self.stamp_detector.model.to(cfg.DEVICE)
 
     def run(self, image):
         # ratio = 1.0*self.standard_width/image.shape[1]