diff --git a/src/remove_ai_watermarks/invisible_engine.py b/src/remove_ai_watermarks/invisible_engine.py index 47d421c..ac33715 100644 --- a/src/remove_ai_watermarks/invisible_engine.py +++ b/src/remove_ai_watermarks/invisible_engine.py @@ -131,8 +131,9 @@ class InvisibleEngine: new_size = (int(image.width * ratio), int(image.height * ratio)) if self._progress_callback: self._progress_callback( - f"Auto-downscaling {image.width}x{image.height} " - f"to {new_size[0]}x{new_size[1]} to prevent Memory Error..." + f"Downscaling {image.width}x{image.height} " + f"to {new_size[0]}x{new_size[1]} " + f"(model trained at {max_dimension}px)..." ) image = image.resize(new_size, Image.Resampling.LANCZOS)