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 |
-
|
| 25 |
from lbm.inference import evaluate, get_model
|
| 26 |
|
| 27 |
# Load model
|
|
|
|
| 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
|