diff --git a/run.py b/run.py
index c59b8bb..36c13e6 100644
--- a/run.py
+++ b/run.py
@@ -185,7 +185,7 @@ def __call__(self, data: AugInput, device: Optional[str] = None) -> tuple[dict,
         Run the model on the image with preprocessing
 
         Args:
-            data (AugInput): image to run the model on
+            data (AugInput): image (and possibly dpi) to run the model on
             device (str): device to run the model on
         Returns:
             tuple[dict, int, int]: predictions, height, width