pearisli commited on
Commit
c792bde
·
verified ·
1 Parent(s): 9ba1662

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +25 -5
README.md CHANGED
@@ -1,9 +1,29 @@
1
  ---
 
2
  tags:
3
- - model_hub_mixin
4
- - pytorch_model_hub_mixin
 
 
5
  ---
6
 
7
- This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
8
- - Library: [More Information Needed]
9
- - Docs: [More Information Needed]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ license: mit
3
  tags:
4
+ - deepdanbooru
5
+ - image-tagging
6
+ - safetensors
7
+ - pytorch
8
  ---
9
 
10
+ # Deepdanbooru-PyTorch Model Card
11
+
12
+ Deepdanbooru-pytorch is a modular, weight-compatible rewrite of [AUTOMATIC1111/TorchDeepDanbooru](https://github.com/AUTOMATIC1111/TorchDeepDanbooru), with all original weights migrated into a `model.safetensors` file for efficient loading.
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")
25
+ model = model.to("cuda")
26
+
27
+ image = Image.open("example.jpg").convert("RGB")
28
+ model.tag(image)
29
+ ```