prithivMLmods's picture
Update README.md
6f6f038 verified
|
Raw
History Blame Contribute Delete
12.2 kB
---
license: apache-2.0
language:
- en
base_model:
- Qwen/Qwen3.5-0.8B
datasets:
- prithivMLmods/ImageShield-Guardrail-Multimodal-20K
pipeline_tag: image-text-to-text
library_name: transformers
tags:
- text-generation-inference
- JSON
- image-classification
- content-safety
- content-moderation
- multimodal
- safety-classifier
- guardrail
- ncii-detection
- visual-safety
- qwen3.5
- multimodal-content-filter
---
![1](https://cdn-uploads.huggingface.co/production/uploads/65bb837dbfb878f46c77de4c/dg6sQtCGMLx-Oosy7-Izq.png)
# **ImageShield-MMCF-0.8B**
> **ImageShield-MMCF — Multimodal Content Filter** is a multimodal content-safety classifier built on top of **Qwen/Qwen3.5-0.8B** and trained on approximately **28,000 content-safety guardrail samples**. The model is designed to classify visual content as **Safe** or **Unsafe**, with a particular focus on detecting **Not Safe for Work (NSFW) sensual content** and other potentially sensitive visual content.
<a href="#example-cards">
<img src="https://img.shields.io/badge/View-Example_Cards-blue?style=for-the-badge" alt="Jump to Examples" />
</a>
> [!NOTE]
> **This model is an experimental release and may generate unexpected behaviors or reasoning artifacts in certain scenarios.**
> [!IMPORTANT]
> **This model is intended for research and content-safety classification only.** Any outputs generated by this model are the sole responsibility of the user. The authors and hosting platform disclaim all liability for generated content. Users must ensure safe, ethical, and lawful usage.
## **Key Highlights**
* **Qwen 3.5 Foundation**: Built on top of **Qwen/Qwen3.5-0.8B**.
* **Multimodal Content Filter**: Designed for visual content-safety classification.
* **28K Training Samples**: Trained on approximately **28,000 content-safety guardrail samples**.
* **Safe / Unsafe Classification**: Classifies visual content as either **Safe** or **Unsafe**.
* **NSFW Sensual Content Detection**: Focused on identifying potentially **Not Safe for Work (NSFW) sensual content** and related sensitive visual material.
* **Zero-Tolerance Safety Classification**: Uses a conservative classification policy that favors safety when the content is ambiguous.
* **Structured Output**: Produces a machine-readable JSON response containing a caption, classification, and reasoning.
## **Standard Instruction**
```text
You are a strict, zero-tolerance parental control content-safety guardrail classifier for visual media.
Your role is minor protection: any media containing suggestive, revealing, sexually provocative, or explicit visual elements must be flagged as NSFW.
STRICT ZERO-TOLERANCE CLASSIFICATION RULES:
1. Nudity & Exposure: Any full or partial nudity, exposed breasts, cleavage, buttocks, groin, or visible pubic area MUST be classified as NSFW.
2. Underwear & Swimwear: Thongs, G-strings, micro-bikinis, skimpy or revealing swimwear, bras, panties, lingerie, see-through clothing, or unbuttoned tops revealing undergarments or bare torsos MUST be classified as NSFW.
3. Poses & Suggestiveness: Sexually suggestive poses, seductive modeling, cleavage-emphasizing framing, fetish content, or explicit erotic themes MUST be classified as NSFW.
4. Artwork & Animation: Anime nudes, ecchi, hentai, suggestive illustrations, or 2D/3D stylized erotica MUST be classified as NSFW.
5. Classification Threshold: When in doubt, err on the side of caution and classify the content as NSFW.
Output strictly valid JSON with no extra conversational text, commentary, or markdown formatting outside the JSON object:
{
"caption": "<Detailed, objective visual description of the subject, clothing, exposure, and pose>",
"is_nsfw": true | false,
"reason": "<Precise reason for classification based on clothing, exposure, or pose>",
"nsfw": 1 | 0,
"safe": 1 | 0
}
```
## **Quick Start with Transformers**
```bash
pip install transformers
pip install accelerate
```
```python
from transformers import Qwen3_5ForConditionalGeneration, AutoProcessor
import torch
model = Qwen3_5ForConditionalGeneration.from_pretrained(
"prithivMLmods/ImageShield-MMCF-0.8B",
torch_dtype="auto",
device_map="auto"
)
processor = AutoProcessor.from_pretrained(
"prithivMLmods/ImageShield-MMCF-0.8B"
)
messages = [
{
"role": "user",
"content": [
{
"type": "text",
"text": "Classify this image using the provided content-safety guardrail."
}
],
}
]
text = processor.apply_chat_template(
messages,
tokenize=False,
add_generation_prompt=True
)
inputs = processor(
text=[text],
padding=True,
return_tensors="pt"
).to("cuda")
generated_ids = model.generate(
**inputs,
max_new_tokens=256
)
output_text = processor.batch_decode(
[
out[len(inp):]
for inp, out in zip(inputs.input_ids, generated_ids)
],
skip_special_tokens=True,
clean_up_tokenization_spaces=False
)
print(output_text[0])
```
---
## **Example Cards**
<div style="background-color: #000000; font-family: 'Courier New', Courier, monospace; color: #FFFFFF; padding: 24px; border-radius: 12px; border: 1px solid #333333; max-width: 800px; margin: 0 auto;">
<!-- Card 1 -->
<div style="margin-bottom: 32px;">
<div style="border-radius: 8px; overflow: hidden; margin-bottom: 16px; border: 1px solid #222;">
<img src="https://cdn-uploads.huggingface.co/production/uploads/65bb837dbfb878f46c77de4c/yypVvKdEH3W5qcmt4E36R.jpeg" alt="Safe Image 1" style="width: 100%; height: auto; display: block;">
</div>
<pre style="background-color: #0a0a0a; padding: 16px; border-radius: 8px; border: 1px solid #222; color: #e0e0e0; font-size: 13px; line-height: 1.6; white-space: pre-wrap; word-wrap: break-word; margin: 0;">
{
"caption": "A woman in a white tank top stands outdoors, gesturing with her right hand near her face while speaking. She wears a silver necklace and a bracelet. To her right, a man in a dark shirt is partially visible, looking towards her. The background shows blurred pedestrians and urban buildings.",
"is_nsfw": <span style="color: #4ade80;">false</span>,
"reason": "The image depicts a casual outdoor scene featuring two adults in everyday attire. There is no evidence of nudity, revealing undergarments, suggestive poses, or any sexually explicit content. The focus is on the interaction between the subjects.",
"nsfw": <span style="color: #4ade80;">0</span>,
"safe": <span style="color: #4ade80;">1</span>
}</pre>
</div>
<!-- Thin Line Bar -->
<hr style="border: 0; border-top: 1px solid #333333; margin: 0 0 32px 0;">
<!-- Card 2 (NSFW - Blurred) -->
<div style="margin-bottom: 32px;">
<div style="border-radius: 8px; overflow: hidden; margin-bottom: 16px; border: 1px solid #222; position: relative;">
<img src="https://cdn-uploads.huggingface.co/production/uploads/65bb837dbfb878f46c77de4c/XI2Xc5WmZrxKTSkVxV13U.png" alt="NSFW Image 2" style="width: 100%; height: auto; display: block; filter: blur(40px) saturate(0.8) brightness(0.9); transform: scale(1.1);">
</div>
<pre style="background-color: #0a0a0a; padding: 16px; border-radius: 8px; border: 1px solid #222; color: #e0e0e0; font-size: 13px; line-height: 1.6; white-space: pre-wrap; word-wrap: break-word; margin: 0;">
{
"caption": "A blonde woman is shown from behind, wearing red thong-style underwear. She has long hair and a visible tattoo on her lower back. Her hands are positioned near her hips, holding the straps of the thong. The setting appears to be a modern living room with a window and a patterned sofa.",
"is_nsfw": <span style="color: #4ade80;">true</span>,
"reason": "The image is classified as NSFW due to revealing thong-style underwear, substantial skin exposure, and a suggestive presentation.",
"nsfw": <span style="color: #4ade80;">1</span>,
"safe": <span style="color: #4ade80;">0</span>
}</pre>
</div>
<!-- Thin Line Bar -->
<hr style="border: 0; border-top: 1px solid #333333; margin: 0 0 32px 0;">
<!-- Card 3 -->
<div>
<div style="border-radius: 8px; overflow: hidden; margin-bottom: 16px; border: 1px solid #222;">
<img src="https://cdn-uploads.huggingface.co/production/uploads/65bb837dbfb878f46c77de4c/EX1Zg_-Up4va_44B3dONZ.jpeg" alt="Safe Image 3" style="width: 100%; height: auto; display: block;">
</div>
<pre style="background-color: #0a0a0a; padding: 16px; border-radius: 8px; border: 1px solid #222; color: #e0e0e0; font-size: 13px; line-height: 1.6; white-space: pre-wrap; word-wrap: break-word; margin: 0;">
{
"caption": "A medium-sized lioness with golden-brown fur walks across a grassy enclosure in a zoo setting. She is walking towards the camera, looking directly at the viewer. The background features large rocks, trees, and a thatched-roof structure, likely a viewing platform or shelter. The environment appears clean and well-maintained.",
"is_nsfw": <span style="color: #4ade80;">false</span>,
"reason": "The image depicts a standard outdoor scene featuring a lioness in a zoo environment. There is no evidence of nudity, revealing clothing, suggestive poses, or any sexually explicit content. The focus is entirely on the animal and its natural habitat.",
"nsfw": <span style="color: #4ade80;">0</span>,
"safe": <span style="color: #4ade80;">1</span>
}</pre>
</div>
</div>
---
## **Training Details**
| Setting | Value |
| :------------------------- | :------------------------------------------- |
| **Base Model** | **Qwen/Qwen3.5-0.8B** |
| **Model Type** | **Multimodal Content-Safety Classifier** |
| **Training Samples** | Approximately **28,000** |
| **Training Objective** | Content-safety guardrail classification |
| **Primary Classification** | **Safe / Unsafe** |
| **Safety Focus** | **Not Safe for Work (NSFW) Sensual Content** |
| **Training Framework** | **TRL** |
## **Intended Use**
* **Content Safety Classification**: Classifying visual media as safe or unsafe.
* **NSFW Sensual Content Detection**: Supporting research into automated detection of potentially sensitive and sexually suggestive visual content.
* **Parental Controls**: Building conservative visual content-safety filtering systems.
* **Content Moderation**: Supporting automated safety classification pipelines.
* **Multimodal Safety Research**: Evaluating content-safety behavior in multimodal language models.
* **Guardrail Development**: Researching structured safety classification and filtering workflows.
## **Limitations**
* **Experimental Model**: The model may produce incorrect or inconsistent classifications.
* **False Positives**: Benign content may occasionally be classified as unsafe due to the conservative classification threshold.
* **False Negatives**: Unsafe content may occasionally be missed.
* **Context Sensitivity**: Classification performance depends on image quality, visual context, and the provided instruction.
* **Conservative Policy**: The model intentionally uses a strict classification threshold and may flag content that would not be considered NSFW under less restrictive moderation policies.
* **Automated Classification**: The model should not be treated as a definitive legal or safety determination.
## **Acknowledgements**
* **[Qwen/Qwen3.5-0.8B](https://huggingface.co/Qwen/Qwen3.5-0.8B)**: Base multimodal model used for this project.
* **TRL – [Transformers Reinforcement Learning](https://huggingface.co/docs/trl/en/index)**: TRL is a full-stack library providing tools to train transformer language models with methods including Supervised Fine-Tuning (SFT), Group Relative Policy Optimization (GRPO), Direct Preference Optimization (DPO), Reward Modeling, and more.
* **[Transformers](https://huggingface.co/docs/transformers/en/index)**: Transformers provides state-of-the-art machine learning models for text, computer vision, audio, video, and multimodal tasks, supporting both inference and training.