Update README.md
Browse files
README.md
CHANGED
@@ -13,12 +13,12 @@ Deepdanbooru-pytorch is a modular, weight-compatible rewrite of [TorchDeepDanboo
|
|
13 |
|
14 |
## Model Details
|
15 |
|
16 |
-
The
|
17 |
|
18 |
## Examples
|
19 |
|
20 |
```python
|
21 |
-
|
22 |
from PIL import Image
|
23 |
|
24 |
model = DeepDanbooruModel.from_pretrained("pearisli/deepdanbooru-pytorch")
|
|
|
13 |
|
14 |
## Model Details
|
15 |
|
16 |
+
The complete implementation is available in the [Pearisli/VisGen repository](https://github.com/Pearisli/VisGen/blob/main/models/deepdanbooru.py). You don’t need to clone the entire repo—simply copy the contents of `deepdanbooru.py` and you’re all set.
|
17 |
|
18 |
## Examples
|
19 |
|
20 |
```python
|
21 |
+
# import or define class DeepDanbooruModel
|
22 |
from PIL import Image
|
23 |
|
24 |
model = DeepDanbooruModel.from_pretrained("pearisli/deepdanbooru-pytorch")
|