pearisli commited on
Commit
9817a8f
·
verified ·
1 Parent(s): 3b7619a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -13,12 +13,12 @@ Deepdanbooru-pytorch is a modular, weight-compatible rewrite of [TorchDeepDanboo
13
 
14
  ## Model Details
15
 
16
- The full implementation can be found in [Pearisli/VisGen](https://github.com/Pearisli/VisGen/blob/main/models/deepdanbooru.py).
17
 
18
  ## Examples
19
 
20
  ```python
21
- from models import DeepDanbooruModel
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")