diff --git a/image_classifier/inference.py b/image_classifier/inference.py index 9a74850..1e5e318 100755 --- a/image_classifier/inference.py +++ b/image_classifier/inference.py @@ -16,6 +16,8 @@ class CAPTCHA(): else: self.load_model_from_cfg( config ) + self.model.eval() + self.width = width self.height = height self.transform = transforms.Compose([