Byt5 Small NSFW Image Urls

🚀 バイトエンコーダ付きT5モデルによる不適切画像URL検出

このプロジェクトは、T5ベースのモデルを用いて、不適切な画像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

データセット情報