from setfit import SetFitModel

# Load the model
model = SetFitModel.from_pretrained("setfit-multilingual-e5-nsfw-classifier")

# Run inference
texts = [
    "domain:example.com\ntitle:Weather forecast\ndescription:Sunny skies expected",
    "domain:news.com\ntitle:Breaking news\ndescription:Important developments",
]
predictions = model.predict(texts)
print(predictions)

Input Format

The model expects input text in the following format:

domain:<domain_name>
title:<article_title>
description:<article_description>

Output

The model outputs one of two labels:

adult: Content is flagged as adult/NSFW
safe: Content is safe for work

Limitations and Bias

This model is trained on a specific dataset and may not generalize well to:

Different languages or domains not represented in training data
Content with ambiguous or context-dependent adult themes
Very short or very long text inputs

Citation

If you use this model, please cite SetFit:

bibtex

@article{https://doi.org/10.48550/arxiv.2209.11055,
  doi = {10.48550/ARXIV.2209.11055},
  url = {https://arxiv.org/abs/2209.11055},
  author = {Tunstall, Lewis and Reimers, Nils and Jo, Unso Eun Seo and Bates, Luke and Korat, Daniel and Wasserblat, Moshe and Pereg, Oren},
  title = {Efficient Few-Shot Learning Without Prompts},
  publisher = {arXiv},
  year = {2022},
  copyright = {Creative Commons Attribution 4.0 International}
}

alunadiderot/setfit-multilingual-e5-adult-classifier

SetFit Multilingual E5 NSFW Classifier

Model Details

Training Details

Training Data

Label Distribution

Training Hyperparameters

Training Date

Evaluation Results

Usage

Using SetFit

Input Format

Output

Limitations and Bias

Citation

alunadiderot/
setfit-multilingual-e5-adult-classifier