README.md

---
datasets:
- laion/laion2B-en-joined
---


## Inference

```python
from  transformers import T5ForConditionalGeneration, AutoTokenizer

tokenizer = AutoTokenizer.from_pretrained("google/byt5-small")

model = T5ForConditionalGeneration.from_pretrained("marianna13/byt5-small-NSFW-image-urls")

def get_label(text):
    input_ids = tokenizer(text, return_tensors="pt", padding=True).input_ids
    outputs = model.generate(input_ids)

    label = tokenizer.batch_decode(outputs, skip_special_tokens=True)

    return label

```