diff --git a/README.md b/README.md
index 6365717a..2647736c 100644
--- a/README.md
+++ b/README.md
@@ -492,7 +492,7 @@ sys.path.append(REPO_DIR)
 
 from PIL import Image
 import torch
-from torchvision import transforms
+from torchvision.transforms import v2
 import matplotlib.pyplot as plt
 from matplotlib import colormaps
 from functools import partial