Update README.md
Browse files
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
|
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 =
|
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)
|