---
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)
```