| --- |
| license: mit |
| tags: |
| - deepdanbooru |
| - safetensors |
| - pytorch |
| pipeline_tag: image-to-text |
| --- |
| |
| # Deepdanbooru-PyTorch Model Card |
|
|
| Deepdanbooru-pytorch is a modular, weight-compatible rewrite of [TorchDeepDanbooru](https://github.com/AUTOMATIC1111/TorchDeepDanbooru), with all original weights migrated into a `model.safetensors` file for efficient loading. |
|
|
| ## Model Details |
|
|
| The model implementation is available in the [deepdanbooru.py](https://huggingface.co/pearisli/deepdanbooru-pytorch/blob/main/deepdanbooru.py). |
| ## Examples |
|
|
| ```python |
| # import or define class DeepDanbooruModel |
| from PIL import Image |
| |
| model = DeepDanbooruModel.from_pretrained("pearisli/deepdanbooru-pytorch") |
| model = model.to("cuda") |
| |
| image = Image.open("example.jpg").convert("RGB") |
| model.tag(image) |
| ``` |
