Byt5 Small NSFW Image Urls

🚀 圖像URL內容分類模型

本項目是一個圖像URL內容分類模型，可對圖像URL進行分類，使用特定數據集訓練，提供推理代碼示例。

🚀 快速開始

此模型可用於對圖像URL進行分類，判斷其是否包含不適當內容。

📦 安裝指南

使用此模型前，請確保已安裝transformers庫，可使用以下命令進行安裝：

pip install transformers

💻 使用示例

基礎用法

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