Image-to-Image
Safetensors
clementchadebec commited on
Commit
bbafcde
·
verified ·
1 Parent(s): bbf2f69

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -4
README.md CHANGED
@@ -21,7 +21,7 @@ pip install git+https://github.com/gojasper/LBM.git
21
  Then, you can infer with the model on your input images
22
  ```python
23
  import torch
24
- from diffusers.utils import load_image
25
  from lbm.inference import evaluate, get_model
26
 
27
  # Load model
@@ -32,9 +32,7 @@ model = get_model(
32
  )
33
 
34
  # Load a source image
35
- source_image = load_image(
36
- "https://huggingface.co/jasperai/LBM_relighting/resolve/main/assets/source_image.jpg"
37
- )
38
 
39
  # Perform inference
40
  output_image = evaluate(model, source_image, num_sampling_steps=1)
 
21
  Then, you can infer with the model on your input images
22
  ```python
23
  import torch
24
+ from PIL import Image
25
  from lbm.inference import evaluate, get_model
26
 
27
  # Load model
 
32
  )
33
 
34
  # Load a source image
35
+ source_image = Image.open("your_image")
 
 
36
 
37
  # Perform inference
38
  output_image = evaluate(model, source_image, num_sampling_steps=1)