README.md

---
language: en
license: mit
tags:
- text-generation
- question-answering
- sentence-similarity
- text-classification
- need-ai
base_model: unitary/toxic-bert
pipeline_tag: text-classification
---

# NEED AI - Content Moderation

BERT model for detecting toxic and inappropriate content in user messages and reviews.

## Model Details

- **Base Model**: unitary/toxic-bert
- **Task**: Text Classification
- **Fine-tuned for**: NEED Service Marketplace Platform
- **Language**: English
- **License**: MIT

## Usage

```python
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
# or AutoModelForSequenceClassification for moderation
# or SentenceTransformer for semantic-search

model_name = "yogami9/need-content-moderation"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)

# Example usage
input_text = "Your input here"
inputs = tokenizer(input_text, return_tensors="pt")
outputs = model.generate(**inputs)
result = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(result)
```

## Training Data

Trained on curated datasets specific to the NEED platform's service categories and user interactions.

## Limitations

- Optimized for English language
- Best performance on NEED platform-specific queries
- May require fine-tuning for other domains

## Contact

- **Organization**: NEED Service App
- **Email**: needserviceapp@gmail.com
- **GitHub**: https://github.com/Need-Service-App

## Related Models

All NEED AI models:
- [Category Recommendation](https://huggingface.co/yogami9/need-category-recommendation)
- [Chat Support](https://huggingface.co/yogami9/need-chat-support)
- [Service Description](https://huggingface.co/yogami9/need-service-description)
- [Semantic Search](https://huggingface.co/yogami9/need-semantic-search)
- [Content Moderation](https://huggingface.co/yogami9/need-content-moderation)