Update README.md
Browse files
README.md
CHANGED
|
@@ -58,7 +58,7 @@ You can download the checkpoint at [🤗 PL-Stitch](https://huggingface.co/visur
|
|
| 58 |
```python
|
| 59 |
import torch
|
| 60 |
from PIL import Image
|
| 61 |
-
from
|
| 62 |
|
| 63 |
# Load the pre-trained pl_stitch model
|
| 64 |
pl_stitch = build_model(pretrained_weights = 'your path to the model')
|
|
|
|
| 58 |
```python
|
| 59 |
import torch
|
| 60 |
from PIL import Image
|
| 61 |
+
from build_model import build_model
|
| 62 |
|
| 63 |
# Load the pre-trained pl_stitch model
|
| 64 |
pl_stitch = build_model(pretrained_weights = 'your path to the model')
|