Update README.md

README.md

@@ -21,7 +21,7 @@ pip install git+https://github.com/gojasper/LBM.git
 Then, you can infer with the model on your input images 
 ```python
 import torch
-from diffusers.utils import load_image
+from PIL import Image
 from lbm.inference import evaluate, get_model
 
 # Load model
@@ -32,9 +32,7 @@ model = get_model(
 )
 
 # Load a source image
-source_image = load_image(
-  "https://huggingface.co/jasperai/LBM_relighting/resolve/main/assets/source_image.jpg"
-)
+source_image = Image.open("your_image")
 
 # Perform inference
 output_image = evaluate(model, source_image, num_sampling_steps=1)