Update README.md

README.md

@@ -47,8 +47,10 @@ pipe = DiffusionPipeline.from_pretrained(
 # Inference
 ```py
 from PIL import Image
+from torchvision.transforms import ToTensor
+
 # Load Control Image
-cond = Image.open("examples/libre_flux_control_image.png")
+cond = Image.open("libre_flux_control_image.png")
 cond = cond.resize((1024, 1024))
 
 # Convert PIL image to tensor and move to device with correct dtype