Upload folder using huggingface_hub

.ipynb_checkpoints/README-checkpoint.md

@@ -0,0 +1,396 @@
+---
+license: apache-2.0
+language:
+- en
+base_model:
+- Ultralytics/YOLO11
+tags:
+- yolo
+- yolo11
+- nsfw
+pipeline_tag: object-detection
+---
+
+<p align="center">
+  <img src="https://cdn-uploads.huggingface.co/production/uploads/63d8d8879dfcfa941d4d7cd9/myOju5oXm0Xq_EgVpYsvo.png" alt="Logo">
+</p>
+
+<h1 style="color:red" style="text-align:center;"> 
+  🔞 WARNING: SENSITIVE CONTENT 🔞 
+</h1>
+
+<h2>
+  THIS MEDIA CONTAINS SENSITIVE CONTENT (I.E. NUDITY, VIOLENCE, PROFANITY, PORN) THAT SOME PEOPLE MAY FIND OFFENSIVE. YOU MUST BE 18 OR OLDER TO VIEW THIS CONTENT.
+</h2>
+
+----------
+# EraX-NSFW-V1.0
+A Highly Efficient Model for NSFW Detection. Very effective for **pre-publication image and video control**, or for **limiting children's access to harmful publications**.
+You can either just predict the classes and their boundingboxes or even mask the predicted harmful object(s) or mask the entire image. 
+Please see the deployment codes below.
+
+- **Developed by**:
+  - Phạm Đình Thục (thuc.pd@erax.ai)
+  - Mr. Nguyễn Anh Nguyên (nguyen@erax.ai)
+- **Model version**: v1.0
+- **License**: Apache 2.0
+
+<!-- ## Examples
+![Example 01](./examples/img_1.jpg) -->
+
+## Model Details / Overview
+
+- **Model Architecture**: YOLO11 (nano, small, medium)
+- **Task**: Object Detection (NSFW Detection)
+- **Dataset**: Private datasets (from Internet).
+- **Training set**: 31890 images.
+- **Validation set**: 11538 images.
+- **Classes**: anus, make_love, nipple, penis, vagina.
+
+### Labels
+![Labels](./train/labels.jpeg)
+
+## Training Configuration
+- **Model Weights Files**:
+  - Nano: [`erax_nsfw_yolo11n.pt`](./erax_nsfw_yolo11n.pt) 
+  - Small: [`erax_nsfw_yolo11s.pt`](./erax_nsfw_yolo11s.pt)
+  - Medium: [`erax_nsfw_yolo11m.pt`](./erax_nsfw_yolo11m.pt)
+- **Number of Epochs**: 100
+- **Learning Rate**: 0.01
+- **Batch Size**: 208
+- **Image Size**: 640x640
+- **Training server**: 8 x NVIDIA RTX A4000 (16GB GDDR6)
+- **Training time**: ~10 hours
+
+
+## Evaluation Metrics
+Below are the key metrics from the model evaluation on the validation set:
+comming soon
+<!-- 
+- **Precision**: 0.726
+- **Recall**: 0.68
+- **mAP50**: 0.724
+- **mAP50-95**: 0.434 -->
+
+<!-- | Format                 | Status |  Size (MB) |  metrics/mAP50-95(B)  |  Inference time (ms/im) |   FPS   |
+|:-----------------------|:-------|:-----------|:----------------------|:------------------------|:--------|
+|  PyTorch               |   ✅   |    38.6    |         0.4332        |          16.97          |  58.91  |
+|TorchScript             |	 ✅	  |    77	   |         0.4153	       |          12.09      	 |  82.69  |
+|   ONNX                 |   ✅	  |    76.7	   |         0.4153	       |          103.94      	 |  9.62   |
+| OpenVINO               |	 ❌	  |    -	   |         -  	       |          -         	 |  -      |
+| TensorRT               | 	 ✅	  |    89.6	   |         0.4155	       |          7          	 |  142.92 |
+| CoreML                 |	 ❌	  |    -	   |         -  	       |          -         	 |  -      |
+| TensorFlow SavedModel  | 	 ✅	  |    192.3   |         0.4153	       |          40.19        	 |  24.88  |
+| TensorFlow GraphDef    |	 ✅	  |    76.8	   |         0.4153	       |          36.71        	 |  27.24  |
+| TensorFlow Lite        |	 ❌	  |    -	   |         -  	       |          -         	 |  -      |
+| TensorFlow Edge TPU    |	 ❌	  |    -	   |         -  	       |          -         	 |  -      |
+| TensorFlow.js          |	 ❌	  |    -	   |         -  	       |          -         	 |  -      |
+| PaddlePaddle           | 	 ✅	  |    153.3   |         0.4153	       |          1024.24      	 |  0.98   |
+| NCNN                   |	 ✅   |    76.6	   |         0.4153	       |          187.36       	 |  5.34   | -->
+
+
+
+<!-- <table>
+    <thead>
+        <tr>
+            <th>Format</th>
+            <th width="200">Model</th>
+            <th width="100">Size (MB)</th>
+            <th>metrics/mAP50-95(B)</th>
+            <th>Inference time (ms/im)</th>
+            <th width="100">FPS</th>
+        </tr>
+    </thead>
+    <tbody>
+        <tr>
+            <th rowspan=3>PyTorch</th>
+            <th>erax_nsfw_yolo11n.pt</th>
+            <th>5.20</th>
+            <th>0.3563</th>
+            <th>160.17</th>
+            <th>6</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11s.pt</th>
+            <th>18.30</th>
+            <th>0.4083</th>
+            <th>10.12</th>
+            <th>99</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11m.pt</th>
+            <th>38.60</th>
+            <th>0.4332</th>
+            <th>12.04</th>
+            <th>83</th>
+        </tr>
+        <tr>
+            <th rowspan=3>TorchScript</th>
+            <th>erax_nsfw_yolo11n.pt</th>
+            <th>10.40</th>
+            <th>0.3427</th>
+            <th>75.11</th>
+            <th>13</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11s.pt</th>
+            <th>36.40</th>
+            <th>0.3918</th>
+            <th>4.40</th>
+            <th>227</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11m.pt</th>
+            <th>77.00</th>
+            <th>0.4153</th>
+            <th>6.32</th>
+            <th>158</th>
+        </tr>
+      <tr>
+            <th rowspan=3>ONNX</th>
+            <th>erax_nsfw_yolo11n.pt</th>
+            <th>10.10</th>
+            <th>0.3427</th>
+            <th>154.94</th>
+            <th>6</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11s.pt</th>
+            <th>36.20</th>
+            <th>0.3918</th>
+            <th>9.82</th>
+            <th>102</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11m.pt</th>
+            <th>76.70</th>
+            <th>0.4153</th>
+            <th>16.38</th>
+            <th>61</th>
+        </tr>
+      <tr>
+            <th rowspan=3>TensorRT</th>
+            <th>erax_nsfw_yolo11n.pt</th>
+            <th>15.60</th>
+            <th>0.3426</th>
+            <th>6.25</th>
+            <th>160</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11s.pt</th>
+            <th>47.30</th>
+            <th>0.3918</th>
+            <th>2.62</th>
+            <th>381</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11m.pt</th>
+            <th>92.70</th>
+            <th>0.4154</th>
+            <th>4.94</th>
+            <th>202</th>
+        </tr>
+      <tr>
+            <th rowspan=3>TensorFlow SavedModel</th>
+            <th>erax_nsfw_yolo11n.pt</th>
+            <th>25.90</th>
+            <th>0.3427</th>
+            <th>27.35</th>
+            <th>37</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11s.pt</th>
+            <th>93.30</th>
+            <th>0.3918</th>
+            <th>33.82</th>
+            <th>30</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11m.pt</th>
+            <th>193.40</th>
+            <th>0.4153</th>
+            <th>34.29</th>
+            <th>29</th>
+        </tr>
+      <tr>
+            <th rowspan=3>TensorFlow GraphDef</th>
+            <th>erax_nsfw_yolo11n.pt</th>
+            <th>10.20</th>
+            <th>0.3427</th>
+            <th>26.58</th>
+            <th>38</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11s.pt</th>
+            <th>36.30</th>
+            <th>0.3918</th>
+            <th>31.84</th>
+            <th>31</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11m.pt</th>
+            <th>76.80</th>
+            <th>0.4153</th>
+            <th>31.83</th>
+            <th>31</th>
+        </tr>
+      <tr>
+            <th rowspan=3>PaddlePaddle</th>
+            <th>erax_nsfw_yolo11n.pt</th>
+            <th>20.10</th>
+            <th>0.3427</th>
+            <th>264.66</th>
+            <th>4</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11s.pt</th>
+            <th>72.30</th>
+            <th>0.3918</th>
+            <th>453.69</th>
+            <th>2</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11m.pt</th>
+            <th>153.30</th>
+            <th>0.4153</th>
+            <th>2519.82</th>
+            <th>0</th>
+        </tr>
+    </tbody>
+  <tr>
+            <th rowspan=3>NCNN</th>
+            <th>erax_nsfw_yolo11n.pt</th>
+            <th>10.00</th>
+            <th>0.3427</th>
+            <th>893.43</th>
+            <th>1</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11s.pt</th>
+            <th>36.10</th>
+            <th>0.3918</th>
+            <th>1418.91</th>
+            <th>1</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11m.pt</th>
+            <th>76.60</th>
+            <th>0.4153</th>
+            <th>2647.62</th>
+            <th>0</th>
+        </tr>
+</table>
+ -->
+
+## Training Validation Results
+### Training and Validation Losses
+![Training and Validation Losses](./train/results.png)
+
+### Confusion Matrix
+![Confusion Matrix](./train/confusion_matrix_normalized.png)
+
+
+## Inference
+
+To use the trained model, follow these steps:
+
+1. **Install the necessary packages**:
+```curl
+pip install ultralytics supervision huggingface-hub
+```
+
+2. **Download Pretrained model**:
+```python
+from huggingface_hub import snapshot_download
+snapshot_download(repo_id="erax-ai/EraX-NSFW-V1.0", local_dir="./", force_download=True)
+``` 
+
+3. **Simple Use Case**:
+```python
+from ultralytics import YOLO
+from PIL import Image
+import supervision as sv
+import numpy as np
+
+IOU_THRESHOLD        = 0.3
+CONFIDENCE_THRESHOLD = 0.2
+
+# pretrained_path = "erax_nsfw_yolo11n.pt"
+# pretrained_path = "erax_nsfw_yolo11s.pt"
+pretrained_path = "erax_nsfw_yolo11m.pt"
+
+image_path_list = ["test_images/img_1.jpg", "test_images/img_2.jpg"]
+
+model = YOLO(pretrained_path)
+results = model(image_path_list,
+                  conf=CONFIDENCE_THRESHOLD,
+                  iou=IOU_THRESHOLD
+                )
+
+
+for result in results:
+    annotated_image = result.orig_img.copy()
+    h, w = annotated_image.shape[:2]
+    anchor = h if h > w else w
+
+    detections = sv.Detections.from_ultralytics(result)
+    label_annotator = sv.LabelAnnotator(text_color=sv.Color.BLACK,
+                                        text_position=sv.Position.CENTER,
+                                        text_scale=anchor/1700)
+    
+    pixelate_annotator = sv.PixelateAnnotator(pixel_size=anchor/50)
+    
+    annotated_image = pixelate_annotator.annotate(
+        scene=annotated_image.copy(),
+        detections=detections
+    )
+
+
+    annotated_image = label_annotator.annotate(
+        annotated_image,
+        detections=detections
+    )
+
+    
+    sv.plot_image(annotated_image, size=(10, 10))
+```
+
+## Training
+Scripts for training: https://github.com/EraX-JS-Company/EraX-NSFW-V1.0
+
+## More examples
+
+1. **Example 01**:
+![Example 03](./examples/img_3.jpg)
+
+
+2. **Example 02**:
+![Example 06](./examples/img_6.jpg)
+
+
+3. **Example 03**: SAFEEST for using make_love class as it will cover entire context.
+Without make_love class    |  With make_love class
+:-------------------------:|:-------------------------:
+![](./examples/img_2.jpg)  |  ![](./examples/img_2_make_love.jpg)
+![](./examples/img_4.jpg)  |  ![](./examples/img_4_make_love.jpg)
+![](./examples/img_5.jpg)  |  ![](./examples/img_5_make_love.jpg)
+
+
+
+## Citation 
+If you find our project useful, we would appreciate it if you could star our repository and cite our work as follows:
+```bibtex
+@article{EraX-NSFW-V1.0,
+  author    = {Phạm Đình Thục and
+              Mr. Nguyễn Anh Nguyên and
+              Đoàn Thành Khang and
+              Mr. Trần Hải Khương and
+              Mr. Trương Công Đức and 
+              Phan Nguyễn Tuấn Kha and 
+              Phạm Huỳnh Nhật},
+  title     = {EraX-NSFW-V1.0: A Highly Efficient Model for NSFW Detection},
+  organization={EraX JS Company},
+  year={2024},
+  url={https://huggingface.co/erax-ai/EraX-NSFW-V1.0}
+}
+```

.ipynb_checkpoints/Upload_Huggingface-checkpoint.ipynb

@@ -0,0 +1,127 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 11,
+   "id": "9a35dbb2-4482-4155-86ca-c12a0a56ba7d",
+   "metadata": {
+    "scrolled": true
+   },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "CommitInfo(commit_url='https://huggingface.co/erax-ai/EraX-NSFW-V1.1/commit/2006614e6e27495bf5e48181a63ab8c666850cc4', commit_message='Upload README.md with huggingface_hub', commit_description='', oid='2006614e6e27495bf5e48181a63ab8c666850cc4', pr_url=None, repo_url=RepoUrl('https://huggingface.co/erax-ai/EraX-NSFW-V1.1', endpoint='https://huggingface.co', repo_type='model', repo_id='erax-ai/EraX-NSFW-V1.1'), pr_revision=None, pr_num=None)"
+      ]
+     },
+     "execution_count": 11,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "from huggingface_hub import HfApi\n",
+    "api = HfApi()\n",
+    "api.upload_file(\n",
+    "    path_or_fileobj=\"/home/workspace/lechitai/Anti-NSFW/EraX-Anti-NSFW-V1.1/README.md\",\n",
+    "    path_in_repo=\"README.md\",\n",
+    "    repo_id=\"erax-ai/EraX-NSFW-V1.1\",\n",
+    "    repo_type=\"model\",\n",
+    "    token=\"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\",\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 14,
+   "id": "cbbcd852-16ca-432c-a299-624f0921f51e",
+   "metadata": {},
+   "outputs": [
+    {
+     "ename": "BadRequestError",
+     "evalue": "(Request ID: Root=1-67655840-0814ae860340e8820ada95d5;e670748b-fb79-4a7c-838e-f5f39b39c133)\n\nBad request for commit endpoint:\nIt appears that one or more of your files contain valid Hugging Face secrets, such as tokens or API keys. It is important to keep your secrets private and not expose them in code that is publicly accessible. Read more: https://huggingface.co/docs/hub/spaces-overview#managing-secrets Offending files: - .ipynb_checkpoints/Upload_Huggingface-checkpoint.ipynb (ref: refs/heads/main, token: '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') - .ipynb_checkpoints/Upload_Huggingface-checkpoint.ipynb (ref: refs/heads/main, token: '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')\nIt appears that one or more of your files contain valid Hugging Face secrets, such as tokens or API keys. It is important to keep your secrets private and not expose them in code that is publicly accessible. Read more: https://huggingface.co/docs/hub/spaces-overview#managing-secrets  Offending files:   - .ipynb_checkpoints/Upload_Huggingface-checkpoint.ipynb (ref: refs/heads/main, token: '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')   - .ipynb_checkpoints/Upload_Huggingface-checkpoint.ipynb (ref: refs/heads/main, token: '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')",
+     "output_type": "error",
+     "traceback": [
+      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
+      "\u001b[0;31mHTTPError\u001b[0m                                 Traceback (most recent call last)",
+      "File \u001b[0;32m/usr/local/lib/python3.10/dist-packages/huggingface_hub/utils/_http.py:406\u001b[0m, in \u001b[0;36mhf_raise_for_status\u001b[0;34m(response, endpoint_name)\u001b[0m\n\u001b[1;32m    405\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[0;32m--> 406\u001b[0m     \u001b[43mresponse\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mraise_for_status\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    407\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m HTTPError \u001b[38;5;28;01mas\u001b[39;00m e:\n",
+      "File \u001b[0;32m/usr/local/lib/python3.10/dist-packages/requests/models.py:1024\u001b[0m, in \u001b[0;36mResponse.raise_for_status\u001b[0;34m(self)\u001b[0m\n\u001b[1;32m   1023\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m http_error_msg:\n\u001b[0;32m-> 1024\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m HTTPError(http_error_msg, response\u001b[38;5;241m=\u001b[39m\u001b[38;5;28mself\u001b[39m)\n",
+      "\u001b[0;31mHTTPError\u001b[0m: 400 Client Error: Bad Request for url: https://huggingface.co/api/models/erax-ai/EraX-NSFW-V1.1/commit/main",
+      "\nThe above exception was the direct cause of the following exception:\n",
+      "\u001b[0;31mBadRequestError\u001b[0m                           Traceback (most recent call last)",
+      "Cell \u001b[0;32mIn[14], line 4\u001b[0m\n\u001b[1;32m      1\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mhuggingface_hub\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m HfApi\n\u001b[1;32m      2\u001b[0m api \u001b[38;5;241m=\u001b[39m HfApi()\n\u001b[0;32m----> 4\u001b[0m \u001b[43mapi\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mupload_folder\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m      5\u001b[0m \u001b[43m    \u001b[49m\u001b[43mfolder_path\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[38;5;124;43m/home/workspace/lechitai/Anti-NSFW/EraX-Anti-NSFW-V1.1\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[43m,\u001b[49m\n\u001b[1;32m      6\u001b[0m \u001b[43m    \u001b[49m\u001b[43mrepo_id\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[38;5;124;43merax-ai/EraX-NSFW-V1.1\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[43m,\u001b[49m\n\u001b[1;32m      7\u001b[0m \u001b[43m    \u001b[49m\u001b[43mrepo_type\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[38;5;124;43mmodel\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[43m,\u001b[49m\n\u001b[1;32m      8\u001b[0m \u001b[43m    \u001b[49m\u001b[43mtoken\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[38;5;124;43mhf_UvzfLMvFhPizsWmSPrYqIsAPFLYHDpGbZn\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[43m,\u001b[49m\n\u001b[1;32m      9\u001b[0m \u001b[43m)\u001b[49m\n",
+      "File \u001b[0;32m/usr/local/lib/python3.10/dist-packages/huggingface_hub/utils/_validators.py:114\u001b[0m, in \u001b[0;36mvalidate_hf_hub_args.<locals>._inner_fn\u001b[0;34m(*args, **kwargs)\u001b[0m\n\u001b[1;32m    111\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m check_use_auth_token:\n\u001b[1;32m    112\u001b[0m     kwargs \u001b[38;5;241m=\u001b[39m smoothly_deprecate_use_auth_token(fn_name\u001b[38;5;241m=\u001b[39mfn\u001b[38;5;241m.\u001b[39m\u001b[38;5;18m__name__\u001b[39m, has_token\u001b[38;5;241m=\u001b[39mhas_token, kwargs\u001b[38;5;241m=\u001b[39mkwargs)\n\u001b[0;32m--> 114\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[43mfn\u001b[49m\u001b[43m(\u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43margs\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mkwargs\u001b[49m\u001b[43m)\u001b[49m\n",
+      "File \u001b[0;32m/usr/local/lib/python3.10/dist-packages/huggingface_hub/hf_api.py:1559\u001b[0m, in \u001b[0;36mfuture_compatible.<locals>._inner\u001b[0;34m(self, *args, **kwargs)\u001b[0m\n\u001b[1;32m   1556\u001b[0m     \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mrun_as_future(fn, \u001b[38;5;28mself\u001b[39m, \u001b[38;5;241m*\u001b[39margs, \u001b[38;5;241m*\u001b[39m\u001b[38;5;241m*\u001b[39mkwargs)\n\u001b[1;32m   1558\u001b[0m \u001b[38;5;66;03m# Otherwise, call the function normally\u001b[39;00m\n\u001b[0;32m-> 1559\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[43mfn\u001b[49m\u001b[43m(\u001b[49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43margs\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mkwargs\u001b[49m\u001b[43m)\u001b[49m\n",
+      "File \u001b[0;32m/usr/local/lib/python3.10/dist-packages/huggingface_hub/hf_api.py:5105\u001b[0m, in \u001b[0;36mHfApi.upload_folder\u001b[0;34m(self, repo_id, folder_path, path_in_repo, commit_message, commit_description, token, repo_type, revision, create_pr, parent_commit, allow_patterns, ignore_patterns, delete_patterns, multi_commits, multi_commits_verbose, run_as_future)\u001b[0m\n\u001b[1;32m   5101\u001b[0m     \u001b[38;5;66;03m# Defining a CommitInfo object is not really relevant in this case\u001b[39;00m\n\u001b[1;32m   5102\u001b[0m     \u001b[38;5;66;03m# Let's return early with pr_url only (as string).\u001b[39;00m\n\u001b[1;32m   5103\u001b[0m     \u001b[38;5;28;01mreturn\u001b[39;00m pr_url\n\u001b[0;32m-> 5105\u001b[0m commit_info \u001b[38;5;241m=\u001b[39m \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mcreate_commit\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m   5106\u001b[0m \u001b[43m    \u001b[49m\u001b[43mrepo_type\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrepo_type\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m   5107\u001b[0m \u001b[43m    \u001b[49m\u001b[43mrepo_id\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrepo_id\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m   5108\u001b[0m \u001b[43m    \u001b[49m\u001b[43moperations\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mcommit_operations\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m   5109\u001b[0m \u001b[43m    \u001b[49m\u001b[43mcommit_message\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mcommit_message\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m   5110\u001b[0m \u001b[43m    \u001b[49m\u001b[43mcommit_description\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mcommit_description\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m   5111\u001b[0m \u001b[43m    \u001b[49m\u001b[43mtoken\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mtoken\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m   5112\u001b[0m \u001b[43m    \u001b[49m\u001b[43mrevision\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrevision\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m   5113\u001b[0m \u001b[43m    \u001b[49m\u001b[43mcreate_pr\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mcreate_pr\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m   5114\u001b[0m \u001b[43m    \u001b[49m\u001b[43mparent_commit\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mparent_commit\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m   5115\u001b[0m \u001b[43m\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m   5117\u001b[0m \u001b[38;5;66;03m# Create url to uploaded folder (for legacy return value)\u001b[39;00m\n\u001b[1;32m   5118\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m create_pr \u001b[38;5;129;01mand\u001b[39;00m commit_info\u001b[38;5;241m.\u001b[39mpr_url \u001b[38;5;129;01mis\u001b[39;00m \u001b[38;5;129;01mnot\u001b[39;00m \u001b[38;5;28;01mNone\u001b[39;00m:\n",
+      "File \u001b[0;32m/usr/local/lib/python3.10/dist-packages/huggingface_hub/utils/_validators.py:114\u001b[0m, in \u001b[0;36mvalidate_hf_hub_args.<locals>._inner_fn\u001b[0;34m(*args, **kwargs)\u001b[0m\n\u001b[1;32m    111\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m check_use_auth_token:\n\u001b[1;32m    112\u001b[0m     kwargs \u001b[38;5;241m=\u001b[39m smoothly_deprecate_use_auth_token(fn_name\u001b[38;5;241m=\u001b[39mfn\u001b[38;5;241m.\u001b[39m\u001b[38;5;18m__name__\u001b[39m, has_token\u001b[38;5;241m=\u001b[39mhas_token, kwargs\u001b[38;5;241m=\u001b[39mkwargs)\n\u001b[0;32m--> 114\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[43mfn\u001b[49m\u001b[43m(\u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43margs\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mkwargs\u001b[49m\u001b[43m)\u001b[49m\n",
+      "File \u001b[0;32m/usr/local/lib/python3.10/dist-packages/huggingface_hub/hf_api.py:1559\u001b[0m, in \u001b[0;36mfuture_compatible.<locals>._inner\u001b[0;34m(self, *args, **kwargs)\u001b[0m\n\u001b[1;32m   1556\u001b[0m     \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mrun_as_future(fn, \u001b[38;5;28mself\u001b[39m, \u001b[38;5;241m*\u001b[39margs, \u001b[38;5;241m*\u001b[39m\u001b[38;5;241m*\u001b[39mkwargs)\n\u001b[1;32m   1558\u001b[0m \u001b[38;5;66;03m# Otherwise, call the function normally\u001b[39;00m\n\u001b[0;32m-> 1559\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[43mfn\u001b[49m\u001b[43m(\u001b[49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43margs\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mkwargs\u001b[49m\u001b[43m)\u001b[49m\n",
+      "File \u001b[0;32m/usr/local/lib/python3.10/dist-packages/huggingface_hub/hf_api.py:4093\u001b[0m, in \u001b[0;36mHfApi.create_commit\u001b[0;34m(self, repo_id, operations, commit_message, commit_description, token, repo_type, revision, create_pr, num_threads, parent_commit, run_as_future)\u001b[0m\n\u001b[1;32m   4091\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[1;32m   4092\u001b[0m     commit_resp \u001b[38;5;241m=\u001b[39m get_session()\u001b[38;5;241m.\u001b[39mpost(url\u001b[38;5;241m=\u001b[39mcommit_url, headers\u001b[38;5;241m=\u001b[39mheaders, data\u001b[38;5;241m=\u001b[39mdata, params\u001b[38;5;241m=\u001b[39mparams)\n\u001b[0;32m-> 4093\u001b[0m     \u001b[43mhf_raise_for_status\u001b[49m\u001b[43m(\u001b[49m\u001b[43mcommit_resp\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mendpoint_name\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[38;5;124;43mcommit\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[43m)\u001b[49m\n\u001b[1;32m   4094\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m RepositoryNotFoundError \u001b[38;5;28;01mas\u001b[39;00m e:\n\u001b[1;32m   4095\u001b[0m     e\u001b[38;5;241m.\u001b[39mappend_to_message(_CREATE_COMMIT_NO_REPO_ERROR_MESSAGE)\n",
+      "File \u001b[0;32m/usr/local/lib/python3.10/dist-packages/huggingface_hub/utils/_http.py:460\u001b[0m, in \u001b[0;36mhf_raise_for_status\u001b[0;34m(response, endpoint_name)\u001b[0m\n\u001b[1;32m    456\u001b[0m \u001b[38;5;28;01melif\u001b[39;00m response\u001b[38;5;241m.\u001b[39mstatus_code \u001b[38;5;241m==\u001b[39m \u001b[38;5;241m400\u001b[39m:\n\u001b[1;32m    457\u001b[0m     message \u001b[38;5;241m=\u001b[39m (\n\u001b[1;32m    458\u001b[0m         \u001b[38;5;124mf\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;124mBad request for \u001b[39m\u001b[38;5;132;01m{\u001b[39;00mendpoint_name\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m endpoint:\u001b[39m\u001b[38;5;124m\"\u001b[39m \u001b[38;5;28;01mif\u001b[39;00m endpoint_name \u001b[38;5;129;01mis\u001b[39;00m \u001b[38;5;129;01mnot\u001b[39;00m \u001b[38;5;28;01mNone\u001b[39;00m \u001b[38;5;28;01melse\u001b[39;00m \u001b[38;5;124m\"\u001b[39m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;124mBad request:\u001b[39m\u001b[38;5;124m\"\u001b[39m\n\u001b[1;32m    459\u001b[0m     )\n\u001b[0;32m--> 460\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m _format(BadRequestError, message, response) \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01me\u001b[39;00m\n\u001b[1;32m    462\u001b[0m \u001b[38;5;28;01melif\u001b[39;00m response\u001b[38;5;241m.\u001b[39mstatus_code \u001b[38;5;241m==\u001b[39m \u001b[38;5;241m403\u001b[39m:\n\u001b[1;32m    463\u001b[0m     message \u001b[38;5;241m=\u001b[39m (\n\u001b[1;32m    464\u001b[0m         \u001b[38;5;124mf\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;132;01m{\u001b[39;00mresponse\u001b[38;5;241m.\u001b[39mstatus_code\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m Forbidden: \u001b[39m\u001b[38;5;132;01m{\u001b[39;00merror_message\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m.\u001b[39m\u001b[38;5;124m\"\u001b[39m\n\u001b[1;32m    465\u001b[0m         \u001b[38;5;241m+\u001b[39m \u001b[38;5;124mf\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;124mCannot access content at: \u001b[39m\u001b[38;5;132;01m{\u001b[39;00mresponse\u001b[38;5;241m.\u001b[39murl\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m.\u001b[39m\u001b[38;5;124m\"\u001b[39m\n\u001b[1;32m    466\u001b[0m         \u001b[38;5;241m+\u001b[39m \u001b[38;5;124m\"\u001b[39m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;124mMake sure your token has the correct permissions.\u001b[39m\u001b[38;5;124m\"\u001b[39m\n\u001b[1;32m    467\u001b[0m     )\n",
+      "\u001b[0;31mBadRequestError\u001b[0m: (Request ID: Root=1-67655840-0814ae860340e8820ada95d5;e670748b-fb79-4a7c-838e-f5f39b39c133)\n\nBad request for commit endpoint:\nIt appears that one or more of your files contain valid Hugging Face secrets, such as tokens or API keys. It is important to keep your secrets private and not expose them in code that is publicly accessible. Read more: https://huggingface.co/docs/hub/spaces-overview#managing-secrets Offending files: - .ipynb_checkpoints/Upload_Huggingface-checkpoint.ipynb (ref: refs/heads/main, token: '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') - .ipynb_checkpoints/Upload_Huggingface-checkpoint.ipynb (ref: refs/heads/main, token: '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')\nIt appears that one or more of your files contain valid Hugging Face secrets, such as tokens or API keys. It is important to keep your secrets private and not expose them in code that is publicly accessible. Read more: https://huggingface.co/docs/hub/spaces-overview#managing-secrets  Offending files:   - .ipynb_checkpoints/Upload_Huggingface-checkpoint.ipynb (ref: refs/heads/main, token: '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')   - .ipynb_checkpoints/Upload_Huggingface-checkpoint.ipynb (ref: refs/heads/main, token: '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')"
+     ]
+    }
+   ],
+   "source": [
+    "from huggingface_hub import HfApi\n",
+    "api = HfApi()\n",
+    "\n",
+    "api.upload_folder(\n",
+    "    folder_path=\"/home/workspace/lechitai/Anti-NSFW/EraX-Anti-NSFW-V1.1\",\n",
+    "    repo_id=\"erax-ai/EraX-NSFW-V1.1\",\n",
+    "    repo_type=\"model\",\n",
+    "    token=\"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\",\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "id": "1aefb2bf-4f5e-4a4f-867e-3401483fd976",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "'/home/workspace/lechitai/Anti-NSFW'"
+      ]
+     },
+     "execution_count": 3,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "pwd"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "bf9486c5-3d8b-4420-bf20-454e427570fd",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.15"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

README.md

@@ -1,3 +1,396 @@
----
-license: apache-2.0
----
+---
+license: apache-2.0
+language:
+- en
+base_model:
+- Ultralytics/YOLO11
+tags:
+- yolo
+- yolo11
+- nsfw
+pipeline_tag: object-detection
+---
+
+<p align="center">
+  <img src="https://cdn-uploads.huggingface.co/production/uploads/63d8d8879dfcfa941d4d7cd9/myOju5oXm0Xq_EgVpYsvo.png" alt="Logo">
+</p>
+
+<h1 style="color:red" style="text-align:center;"> 
+  🔞 WARNING: SENSITIVE CONTENT 🔞 
+</h1>
+
+<h2>
+  THIS MEDIA CONTAINS SENSITIVE CONTENT (I.E. NUDITY, VIOLENCE, PROFANITY, PORN) THAT SOME PEOPLE MAY FIND OFFENSIVE. YOU MUST BE 18 OR OLDER TO VIEW THIS CONTENT.
+</h2>
+
+----------
+# EraX-NSFW-V1.0
+A Highly Efficient Model for NSFW Detection. Very effective for **pre-publication image and video control**, or for **limiting children's access to harmful publications**.
+You can either just predict the classes and their boundingboxes or even mask the predicted harmful object(s) or mask the entire image. 
+Please see the deployment codes below.
+
+- **Developed by**:
+  - Phạm Đình Thục (thuc.pd@erax.ai)
+  - Mr. Nguyễn Anh Nguyên (nguyen@erax.ai)
+- **Model version**: v1.0
+- **License**: Apache 2.0
+
+<!-- ## Examples
+![Example 01](./examples/img_1.jpg) -->
+
+## Model Details / Overview
+
+- **Model Architecture**: YOLO11 (nano, small, medium)
+- **Task**: Object Detection (NSFW Detection)
+- **Dataset**: Private datasets (from Internet).
+- **Training set**: 31890 images.
+- **Validation set**: 11538 images.
+- **Classes**: anus, make_love, nipple, penis, vagina.
+
+### Labels
+![Labels](./train/labels.jpeg)
+
+## Training Configuration
+- **Model Weights Files**:
+  - Nano: [`erax_nsfw_yolo11n.pt`](./erax_nsfw_yolo11n.pt) 
+  - Small: [`erax_nsfw_yolo11s.pt`](./erax_nsfw_yolo11s.pt)
+  - Medium: [`erax_nsfw_yolo11m.pt`](./erax_nsfw_yolo11m.pt)
+- **Number of Epochs**: 100
+- **Learning Rate**: 0.01
+- **Batch Size**: 208
+- **Image Size**: 640x640
+- **Training server**: 8 x NVIDIA RTX A4000 (16GB GDDR6)
+- **Training time**: ~10 hours
+
+
+## Evaluation Metrics
+Below are the key metrics from the model evaluation on the validation set:
+comming soon
+<!-- 
+- **Precision**: 0.726
+- **Recall**: 0.68
+- **mAP50**: 0.724
+- **mAP50-95**: 0.434 -->
+
+<!-- | Format                 | Status |  Size (MB) |  metrics/mAP50-95(B)  |  Inference time (ms/im) |   FPS   |
+|:-----------------------|:-------|:-----------|:----------------------|:------------------------|:--------|
+|  PyTorch               |   ✅   |    38.6    |         0.4332        |          16.97          |  58.91  |
+|TorchScript             |	 ✅	  |    77	   |         0.4153	       |          12.09      	 |  82.69  |
+|   ONNX                 |   ✅	  |    76.7	   |         0.4153	       |          103.94      	 |  9.62   |
+| OpenVINO               |	 ❌	  |    -	   |         -  	       |          -         	 |  -      |
+| TensorRT               | 	 ✅	  |    89.6	   |         0.4155	       |          7          	 |  142.92 |
+| CoreML                 |	 ❌	  |    -	   |         -  	       |          -         	 |  -      |
+| TensorFlow SavedModel  | 	 ✅	  |    192.3   |         0.4153	       |          40.19        	 |  24.88  |
+| TensorFlow GraphDef    |	 ✅	  |    76.8	   |         0.4153	       |          36.71        	 |  27.24  |
+| TensorFlow Lite        |	 ❌	  |    -	   |         -  	       |          -         	 |  -      |
+| TensorFlow Edge TPU    |	 ❌	  |    -	   |         -  	       |          -         	 |  -      |
+| TensorFlow.js          |	 ❌	  |    -	   |         -  	       |          -         	 |  -      |
+| PaddlePaddle           | 	 ✅	  |    153.3   |         0.4153	       |          1024.24      	 |  0.98   |
+| NCNN                   |	 ✅   |    76.6	   |         0.4153	       |          187.36       	 |  5.34   | -->
+
+
+
+<!-- <table>
+    <thead>
+        <tr>
+            <th>Format</th>
+            <th width="200">Model</th>
+            <th width="100">Size (MB)</th>
+            <th>metrics/mAP50-95(B)</th>
+            <th>Inference time (ms/im)</th>
+            <th width="100">FPS</th>
+        </tr>
+    </thead>
+    <tbody>
+        <tr>
+            <th rowspan=3>PyTorch</th>
+            <th>erax_nsfw_yolo11n.pt</th>
+            <th>5.20</th>
+            <th>0.3563</th>
+            <th>160.17</th>
+            <th>6</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11s.pt</th>
+            <th>18.30</th>
+            <th>0.4083</th>
+            <th>10.12</th>
+            <th>99</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11m.pt</th>
+            <th>38.60</th>
+            <th>0.4332</th>
+            <th>12.04</th>
+            <th>83</th>
+        </tr>
+        <tr>
+            <th rowspan=3>TorchScript</th>
+            <th>erax_nsfw_yolo11n.pt</th>
+            <th>10.40</th>
+            <th>0.3427</th>
+            <th>75.11</th>
+            <th>13</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11s.pt</th>
+            <th>36.40</th>
+            <th>0.3918</th>
+            <th>4.40</th>
+            <th>227</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11m.pt</th>
+            <th>77.00</th>
+            <th>0.4153</th>
+            <th>6.32</th>
+            <th>158</th>
+        </tr>
+      <tr>
+            <th rowspan=3>ONNX</th>
+            <th>erax_nsfw_yolo11n.pt</th>
+            <th>10.10</th>
+            <th>0.3427</th>
+            <th>154.94</th>
+            <th>6</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11s.pt</th>
+            <th>36.20</th>
+            <th>0.3918</th>
+            <th>9.82</th>
+            <th>102</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11m.pt</th>
+            <th>76.70</th>
+            <th>0.4153</th>
+            <th>16.38</th>
+            <th>61</th>
+        </tr>
+      <tr>
+            <th rowspan=3>TensorRT</th>
+            <th>erax_nsfw_yolo11n.pt</th>
+            <th>15.60</th>
+            <th>0.3426</th>
+            <th>6.25</th>
+            <th>160</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11s.pt</th>
+            <th>47.30</th>
+            <th>0.3918</th>
+            <th>2.62</th>
+            <th>381</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11m.pt</th>
+            <th>92.70</th>
+            <th>0.4154</th>
+            <th>4.94</th>
+            <th>202</th>
+        </tr>
+      <tr>
+            <th rowspan=3>TensorFlow SavedModel</th>
+            <th>erax_nsfw_yolo11n.pt</th>
+            <th>25.90</th>
+            <th>0.3427</th>
+            <th>27.35</th>
+            <th>37</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11s.pt</th>
+            <th>93.30</th>
+            <th>0.3918</th>
+            <th>33.82</th>
+            <th>30</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11m.pt</th>
+            <th>193.40</th>
+            <th>0.4153</th>
+            <th>34.29</th>
+            <th>29</th>
+        </tr>
+      <tr>
+            <th rowspan=3>TensorFlow GraphDef</th>
+            <th>erax_nsfw_yolo11n.pt</th>
+            <th>10.20</th>
+            <th>0.3427</th>
+            <th>26.58</th>
+            <th>38</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11s.pt</th>
+            <th>36.30</th>
+            <th>0.3918</th>
+            <th>31.84</th>
+            <th>31</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11m.pt</th>
+            <th>76.80</th>
+            <th>0.4153</th>
+            <th>31.83</th>
+            <th>31</th>
+        </tr>
+      <tr>
+            <th rowspan=3>PaddlePaddle</th>
+            <th>erax_nsfw_yolo11n.pt</th>
+            <th>20.10</th>
+            <th>0.3427</th>
+            <th>264.66</th>
+            <th>4</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11s.pt</th>
+            <th>72.30</th>
+            <th>0.3918</th>
+            <th>453.69</th>
+            <th>2</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11m.pt</th>
+            <th>153.30</th>
+            <th>0.4153</th>
+            <th>2519.82</th>
+            <th>0</th>
+        </tr>
+    </tbody>
+  <tr>
+            <th rowspan=3>NCNN</th>
+            <th>erax_nsfw_yolo11n.pt</th>
+            <th>10.00</th>
+            <th>0.3427</th>
+            <th>893.43</th>
+            <th>1</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11s.pt</th>
+            <th>36.10</th>
+            <th>0.3918</th>
+            <th>1418.91</th>
+            <th>1</th>
+        </tr>
+        <tr>
+            <th>erax_nsfw_yolo11m.pt</th>
+            <th>76.60</th>
+            <th>0.4153</th>
+            <th>2647.62</th>
+            <th>0</th>
+        </tr>
+</table>
+ -->
+
+## Training Validation Results
+### Training and Validation Losses
+![Training and Validation Losses](./train/results.png)
+
+### Confusion Matrix
+![Confusion Matrix](./train/confusion_matrix_normalized.png)
+
+
+## Inference
+
+To use the trained model, follow these steps:
+
+1. **Install the necessary packages**:
+```curl
+pip install ultralytics supervision huggingface-hub
+```
+
+2. **Download Pretrained model**:
+```python
+from huggingface_hub import snapshot_download
+snapshot_download(repo_id="erax-ai/EraX-NSFW-V1.0", local_dir="./", force_download=True)
+``` 
+
+3. **Simple Use Case**:
+```python
+from ultralytics import YOLO
+from PIL import Image
+import supervision as sv
+import numpy as np
+
+IOU_THRESHOLD        = 0.3
+CONFIDENCE_THRESHOLD = 0.2
+
+# pretrained_path = "erax_nsfw_yolo11n.pt"
+# pretrained_path = "erax_nsfw_yolo11s.pt"
+pretrained_path = "erax_nsfw_yolo11m.pt"
+
+image_path_list = ["test_images/img_1.jpg", "test_images/img_2.jpg"]
+
+model = YOLO(pretrained_path)
+results = model(image_path_list,
+                  conf=CONFIDENCE_THRESHOLD,
+                  iou=IOU_THRESHOLD
+                )
+
+
+for result in results:
+    annotated_image = result.orig_img.copy()
+    h, w = annotated_image.shape[:2]
+    anchor = h if h > w else w
+
+    detections = sv.Detections.from_ultralytics(result)
+    label_annotator = sv.LabelAnnotator(text_color=sv.Color.BLACK,
+                                        text_position=sv.Position.CENTER,
+                                        text_scale=anchor/1700)
+    
+    pixelate_annotator = sv.PixelateAnnotator(pixel_size=anchor/50)
+    
+    annotated_image = pixelate_annotator.annotate(
+        scene=annotated_image.copy(),
+        detections=detections
+    )
+
+
+    annotated_image = label_annotator.annotate(
+        annotated_image,
+        detections=detections
+    )
+
+    
+    sv.plot_image(annotated_image, size=(10, 10))
+```
+
+## Training
+Scripts for training: https://github.com/EraX-JS-Company/EraX-NSFW-V1.0
+
+## More examples
+
+1. **Example 01**:
+![Example 03](./examples/img_3.jpg)
+
+
+2. **Example 02**:
+![Example 06](./examples/img_6.jpg)
+
+
+3. **Example 03**: SAFEEST for using make_love class as it will cover entire context.
+Without make_love class    |  With make_love class
+:-------------------------:|:-------------------------:
+![](./examples/img_2.jpg)  |  ![](./examples/img_2_make_love.jpg)
+![](./examples/img_4.jpg)  |  ![](./examples/img_4_make_love.jpg)
+![](./examples/img_5.jpg)  |  ![](./examples/img_5_make_love.jpg)
+
+
+
+## Citation 
+If you find our project useful, we would appreciate it if you could star our repository and cite our work as follows:
+```bibtex
+@article{EraX-NSFW-V1.0,
+  author    = {Phạm Đình Thục and
+              Mr. Nguyễn Anh Nguyên and
+              Đoàn Thành Khang and
+              Mr. Trần Hải Khương and
+              Mr. Trương Công Đức and 
+              Phan Nguyễn Tuấn Kha and 
+              Phạm Huỳnh Nhật},
+  title     = {EraX-NSFW-V1.0: A Highly Efficient Model for NSFW Detection},
+  organization={EraX JS Company},
+  year={2024},
+  url={https://huggingface.co/erax-ai/EraX-NSFW-V1.0}
+}
+```

Upload_Huggingface.ipynb

@@ -0,0 +1,127 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 11,
+   "id": "9a35dbb2-4482-4155-86ca-c12a0a56ba7d",
+   "metadata": {
+    "scrolled": true
+   },
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "CommitInfo(commit_url='https://huggingface.co/erax-ai/EraX-NSFW-V1.1/commit/2006614e6e27495bf5e48181a63ab8c666850cc4', commit_message='Upload README.md with huggingface_hub', commit_description='', oid='2006614e6e27495bf5e48181a63ab8c666850cc4', pr_url=None, repo_url=RepoUrl('https://huggingface.co/erax-ai/EraX-NSFW-V1.1', endpoint='https://huggingface.co', repo_type='model', repo_id='erax-ai/EraX-NSFW-V1.1'), pr_revision=None, pr_num=None)"
+      ]
+     },
+     "execution_count": 11,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "from huggingface_hub import HfApi\n",
+    "api = HfApi()\n",
+    "api.upload_file(\n",
+    "    path_or_fileobj=\"/home/workspace/lechitai/Anti-NSFW/EraX-Anti-NSFW-V1.1/README.md\",\n",
+    "    path_in_repo=\"README.md\",\n",
+    "    repo_id=\"erax-ai/EraX-NSFW-V1.1\",\n",
+    "    repo_type=\"model\",\n",
+    "    token=\"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\",\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 14,
+   "id": "cbbcd852-16ca-432c-a299-624f0921f51e",
+   "metadata": {},
+   "outputs": [
+    {
+     "ename": "BadRequestError",
+     "evalue": "(Request ID: Root=1-67655840-0814ae860340e8820ada95d5;e670748b-fb79-4a7c-838e-f5f39b39c133)\n\nBad request for commit endpoint:\nIt appears that one or more of your files contain valid Hugging Face secrets, such as tokens or API keys. It is important to keep your secrets private and not expose them in code that is publicly accessible. Read more: https://huggingface.co/docs/hub/spaces-overview#managing-secrets Offending files: - .ipynb_checkpoints/Upload_Huggingface-checkpoint.ipynb (ref: refs/heads/main, token: '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') - .ipynb_checkpoints/Upload_Huggingface-checkpoint.ipynb (ref: refs/heads/main, token: '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')\nIt appears that one or more of your files contain valid Hugging Face secrets, such as tokens or API keys. It is important to keep your secrets private and not expose them in code that is publicly accessible. Read more: https://huggingface.co/docs/hub/spaces-overview#managing-secrets  Offending files:   - .ipynb_checkpoints/Upload_Huggingface-checkpoint.ipynb (ref: refs/heads/main, token: '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')   - .ipynb_checkpoints/Upload_Huggingface-checkpoint.ipynb (ref: refs/heads/main, token: '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')",
+     "output_type": "error",
+     "traceback": [
+      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
+      "\u001b[0;31mHTTPError\u001b[0m                                 Traceback (most recent call last)",
+      "File \u001b[0;32m/usr/local/lib/python3.10/dist-packages/huggingface_hub/utils/_http.py:406\u001b[0m, in \u001b[0;36mhf_raise_for_status\u001b[0;34m(response, endpoint_name)\u001b[0m\n\u001b[1;32m    405\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[0;32m--> 406\u001b[0m     \u001b[43mresponse\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mraise_for_status\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m    407\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m HTTPError \u001b[38;5;28;01mas\u001b[39;00m e:\n",
+      "File \u001b[0;32m/usr/local/lib/python3.10/dist-packages/requests/models.py:1024\u001b[0m, in \u001b[0;36mResponse.raise_for_status\u001b[0;34m(self)\u001b[0m\n\u001b[1;32m   1023\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m http_error_msg:\n\u001b[0;32m-> 1024\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m HTTPError(http_error_msg, response\u001b[38;5;241m=\u001b[39m\u001b[38;5;28mself\u001b[39m)\n",
+      "\u001b[0;31mHTTPError\u001b[0m: 400 Client Error: Bad Request for url: https://huggingface.co/api/models/erax-ai/EraX-NSFW-V1.1/commit/main",
+      "\nThe above exception was the direct cause of the following exception:\n",
+      "\u001b[0;31mBadRequestError\u001b[0m                           Traceback (most recent call last)",
+      "Cell \u001b[0;32mIn[14], line 4\u001b[0m\n\u001b[1;32m      1\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mhuggingface_hub\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m HfApi\n\u001b[1;32m      2\u001b[0m api \u001b[38;5;241m=\u001b[39m HfApi()\n\u001b[0;32m----> 4\u001b[0m \u001b[43mapi\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mupload_folder\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m      5\u001b[0m \u001b[43m    \u001b[49m\u001b[43mfolder_path\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[38;5;124;43m/home/workspace/lechitai/Anti-NSFW/EraX-Anti-NSFW-V1.1\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[43m,\u001b[49m\n\u001b[1;32m      6\u001b[0m \u001b[43m    \u001b[49m\u001b[43mrepo_id\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[38;5;124;43merax-ai/EraX-NSFW-V1.1\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[43m,\u001b[49m\n\u001b[1;32m      7\u001b[0m \u001b[43m    \u001b[49m\u001b[43mrepo_type\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[38;5;124;43mmodel\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[43m,\u001b[49m\n\u001b[1;32m      8\u001b[0m \u001b[43m    \u001b[49m\u001b[43mtoken\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[38;5;124;43mhf_UvzfLMvFhPizsWmSPrYqIsAPFLYHDpGbZn\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[43m,\u001b[49m\n\u001b[1;32m      9\u001b[0m \u001b[43m)\u001b[49m\n",
+      "File \u001b[0;32m/usr/local/lib/python3.10/dist-packages/huggingface_hub/utils/_validators.py:114\u001b[0m, in \u001b[0;36mvalidate_hf_hub_args.<locals>._inner_fn\u001b[0;34m(*args, **kwargs)\u001b[0m\n\u001b[1;32m    111\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m check_use_auth_token:\n\u001b[1;32m    112\u001b[0m     kwargs \u001b[38;5;241m=\u001b[39m smoothly_deprecate_use_auth_token(fn_name\u001b[38;5;241m=\u001b[39mfn\u001b[38;5;241m.\u001b[39m\u001b[38;5;18m__name__\u001b[39m, has_token\u001b[38;5;241m=\u001b[39mhas_token, kwargs\u001b[38;5;241m=\u001b[39mkwargs)\n\u001b[0;32m--> 114\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[43mfn\u001b[49m\u001b[43m(\u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43margs\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mkwargs\u001b[49m\u001b[43m)\u001b[49m\n",
+      "File \u001b[0;32m/usr/local/lib/python3.10/dist-packages/huggingface_hub/hf_api.py:1559\u001b[0m, in \u001b[0;36mfuture_compatible.<locals>._inner\u001b[0;34m(self, *args, **kwargs)\u001b[0m\n\u001b[1;32m   1556\u001b[0m     \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mrun_as_future(fn, \u001b[38;5;28mself\u001b[39m, \u001b[38;5;241m*\u001b[39margs, \u001b[38;5;241m*\u001b[39m\u001b[38;5;241m*\u001b[39mkwargs)\n\u001b[1;32m   1558\u001b[0m \u001b[38;5;66;03m# Otherwise, call the function normally\u001b[39;00m\n\u001b[0;32m-> 1559\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[43mfn\u001b[49m\u001b[43m(\u001b[49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43margs\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mkwargs\u001b[49m\u001b[43m)\u001b[49m\n",
+      "File \u001b[0;32m/usr/local/lib/python3.10/dist-packages/huggingface_hub/hf_api.py:5105\u001b[0m, in \u001b[0;36mHfApi.upload_folder\u001b[0;34m(self, repo_id, folder_path, path_in_repo, commit_message, commit_description, token, repo_type, revision, create_pr, parent_commit, allow_patterns, ignore_patterns, delete_patterns, multi_commits, multi_commits_verbose, run_as_future)\u001b[0m\n\u001b[1;32m   5101\u001b[0m     \u001b[38;5;66;03m# Defining a CommitInfo object is not really relevant in this case\u001b[39;00m\n\u001b[1;32m   5102\u001b[0m     \u001b[38;5;66;03m# Let's return early with pr_url only (as string).\u001b[39;00m\n\u001b[1;32m   5103\u001b[0m     \u001b[38;5;28;01mreturn\u001b[39;00m pr_url\n\u001b[0;32m-> 5105\u001b[0m commit_info \u001b[38;5;241m=\u001b[39m \u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mcreate_commit\u001b[49m\u001b[43m(\u001b[49m\n\u001b[1;32m   5106\u001b[0m \u001b[43m    \u001b[49m\u001b[43mrepo_type\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrepo_type\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m   5107\u001b[0m \u001b[43m    \u001b[49m\u001b[43mrepo_id\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrepo_id\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m   5108\u001b[0m \u001b[43m    \u001b[49m\u001b[43moperations\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mcommit_operations\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m   5109\u001b[0m \u001b[43m    \u001b[49m\u001b[43mcommit_message\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mcommit_message\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m   5110\u001b[0m \u001b[43m    \u001b[49m\u001b[43mcommit_description\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mcommit_description\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m   5111\u001b[0m \u001b[43m    \u001b[49m\u001b[43mtoken\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mtoken\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m   5112\u001b[0m \u001b[43m    \u001b[49m\u001b[43mrevision\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mrevision\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m   5113\u001b[0m \u001b[43m    \u001b[49m\u001b[43mcreate_pr\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mcreate_pr\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m   5114\u001b[0m \u001b[43m    \u001b[49m\u001b[43mparent_commit\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[43mparent_commit\u001b[49m\u001b[43m,\u001b[49m\n\u001b[1;32m   5115\u001b[0m \u001b[43m\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m   5117\u001b[0m \u001b[38;5;66;03m# Create url to uploaded folder (for legacy return value)\u001b[39;00m\n\u001b[1;32m   5118\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m create_pr \u001b[38;5;129;01mand\u001b[39;00m commit_info\u001b[38;5;241m.\u001b[39mpr_url \u001b[38;5;129;01mis\u001b[39;00m \u001b[38;5;129;01mnot\u001b[39;00m \u001b[38;5;28;01mNone\u001b[39;00m:\n",
+      "File \u001b[0;32m/usr/local/lib/python3.10/dist-packages/huggingface_hub/utils/_validators.py:114\u001b[0m, in \u001b[0;36mvalidate_hf_hub_args.<locals>._inner_fn\u001b[0;34m(*args, **kwargs)\u001b[0m\n\u001b[1;32m    111\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m check_use_auth_token:\n\u001b[1;32m    112\u001b[0m     kwargs \u001b[38;5;241m=\u001b[39m smoothly_deprecate_use_auth_token(fn_name\u001b[38;5;241m=\u001b[39mfn\u001b[38;5;241m.\u001b[39m\u001b[38;5;18m__name__\u001b[39m, has_token\u001b[38;5;241m=\u001b[39mhas_token, kwargs\u001b[38;5;241m=\u001b[39mkwargs)\n\u001b[0;32m--> 114\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[43mfn\u001b[49m\u001b[43m(\u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43margs\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mkwargs\u001b[49m\u001b[43m)\u001b[49m\n",
+      "File \u001b[0;32m/usr/local/lib/python3.10/dist-packages/huggingface_hub/hf_api.py:1559\u001b[0m, in \u001b[0;36mfuture_compatible.<locals>._inner\u001b[0;34m(self, *args, **kwargs)\u001b[0m\n\u001b[1;32m   1556\u001b[0m     \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[38;5;28mself\u001b[39m\u001b[38;5;241m.\u001b[39mrun_as_future(fn, \u001b[38;5;28mself\u001b[39m, \u001b[38;5;241m*\u001b[39margs, \u001b[38;5;241m*\u001b[39m\u001b[38;5;241m*\u001b[39mkwargs)\n\u001b[1;32m   1558\u001b[0m \u001b[38;5;66;03m# Otherwise, call the function normally\u001b[39;00m\n\u001b[0;32m-> 1559\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m \u001b[43mfn\u001b[49m\u001b[43m(\u001b[49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43margs\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[38;5;241;43m*\u001b[39;49m\u001b[43mkwargs\u001b[49m\u001b[43m)\u001b[49m\n",
+      "File \u001b[0;32m/usr/local/lib/python3.10/dist-packages/huggingface_hub/hf_api.py:4093\u001b[0m, in \u001b[0;36mHfApi.create_commit\u001b[0;34m(self, repo_id, operations, commit_message, commit_description, token, repo_type, revision, create_pr, num_threads, parent_commit, run_as_future)\u001b[0m\n\u001b[1;32m   4091\u001b[0m \u001b[38;5;28;01mtry\u001b[39;00m:\n\u001b[1;32m   4092\u001b[0m     commit_resp \u001b[38;5;241m=\u001b[39m get_session()\u001b[38;5;241m.\u001b[39mpost(url\u001b[38;5;241m=\u001b[39mcommit_url, headers\u001b[38;5;241m=\u001b[39mheaders, data\u001b[38;5;241m=\u001b[39mdata, params\u001b[38;5;241m=\u001b[39mparams)\n\u001b[0;32m-> 4093\u001b[0m     \u001b[43mhf_raise_for_status\u001b[49m\u001b[43m(\u001b[49m\u001b[43mcommit_resp\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mendpoint_name\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[38;5;124;43mcommit\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[43m)\u001b[49m\n\u001b[1;32m   4094\u001b[0m \u001b[38;5;28;01mexcept\u001b[39;00m RepositoryNotFoundError \u001b[38;5;28;01mas\u001b[39;00m e:\n\u001b[1;32m   4095\u001b[0m     e\u001b[38;5;241m.\u001b[39mappend_to_message(_CREATE_COMMIT_NO_REPO_ERROR_MESSAGE)\n",
+      "File \u001b[0;32m/usr/local/lib/python3.10/dist-packages/huggingface_hub/utils/_http.py:460\u001b[0m, in \u001b[0;36mhf_raise_for_status\u001b[0;34m(response, endpoint_name)\u001b[0m\n\u001b[1;32m    456\u001b[0m \u001b[38;5;28;01melif\u001b[39;00m response\u001b[38;5;241m.\u001b[39mstatus_code \u001b[38;5;241m==\u001b[39m \u001b[38;5;241m400\u001b[39m:\n\u001b[1;32m    457\u001b[0m     message \u001b[38;5;241m=\u001b[39m (\n\u001b[1;32m    458\u001b[0m         \u001b[38;5;124mf\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;124mBad request for \u001b[39m\u001b[38;5;132;01m{\u001b[39;00mendpoint_name\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m endpoint:\u001b[39m\u001b[38;5;124m\"\u001b[39m \u001b[38;5;28;01mif\u001b[39;00m endpoint_name \u001b[38;5;129;01mis\u001b[39;00m \u001b[38;5;129;01mnot\u001b[39;00m \u001b[38;5;28;01mNone\u001b[39;00m \u001b[38;5;28;01melse\u001b[39;00m \u001b[38;5;124m\"\u001b[39m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;124mBad request:\u001b[39m\u001b[38;5;124m\"\u001b[39m\n\u001b[1;32m    459\u001b[0m     )\n\u001b[0;32m--> 460\u001b[0m     \u001b[38;5;28;01mraise\u001b[39;00m _format(BadRequestError, message, response) \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01me\u001b[39;00m\n\u001b[1;32m    462\u001b[0m \u001b[38;5;28;01melif\u001b[39;00m response\u001b[38;5;241m.\u001b[39mstatus_code \u001b[38;5;241m==\u001b[39m \u001b[38;5;241m403\u001b[39m:\n\u001b[1;32m    463\u001b[0m     message \u001b[38;5;241m=\u001b[39m (\n\u001b[1;32m    464\u001b[0m         \u001b[38;5;124mf\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;132;01m{\u001b[39;00mresponse\u001b[38;5;241m.\u001b[39mstatus_code\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m Forbidden: \u001b[39m\u001b[38;5;132;01m{\u001b[39;00merror_message\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m.\u001b[39m\u001b[38;5;124m\"\u001b[39m\n\u001b[1;32m    465\u001b[0m         \u001b[38;5;241m+\u001b[39m \u001b[38;5;124mf\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;124mCannot access content at: \u001b[39m\u001b[38;5;132;01m{\u001b[39;00mresponse\u001b[38;5;241m.\u001b[39murl\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m.\u001b[39m\u001b[38;5;124m\"\u001b[39m\n\u001b[1;32m    466\u001b[0m         \u001b[38;5;241m+\u001b[39m \u001b[38;5;124m\"\u001b[39m\u001b[38;5;130;01m\\n\u001b[39;00m\u001b[38;5;124mMake sure your token has the correct permissions.\u001b[39m\u001b[38;5;124m\"\u001b[39m\n\u001b[1;32m    467\u001b[0m     )\n",
+      "\u001b[0;31mBadRequestError\u001b[0m: (Request ID: Root=1-67655840-0814ae860340e8820ada95d5;e670748b-fb79-4a7c-838e-f5f39b39c133)\n\nBad request for commit endpoint:\nIt appears that one or more of your files contain valid Hugging Face secrets, such as tokens or API keys. It is important to keep your secrets private and not expose them in code that is publicly accessible. Read more: https://huggingface.co/docs/hub/spaces-overview#managing-secrets Offending files: - .ipynb_checkpoints/Upload_Huggingface-checkpoint.ipynb (ref: refs/heads/main, token: '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') - .ipynb_checkpoints/Upload_Huggingface-checkpoint.ipynb (ref: refs/heads/main, token: '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')\nIt appears that one or more of your files contain valid Hugging Face secrets, such as tokens or API keys. It is important to keep your secrets private and not expose them in code that is publicly accessible. Read more: https://huggingface.co/docs/hub/spaces-overview#managing-secrets  Offending files:   - .ipynb_checkpoints/Upload_Huggingface-checkpoint.ipynb (ref: refs/heads/main, token: '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')   - .ipynb_checkpoints/Upload_Huggingface-checkpoint.ipynb (ref: refs/heads/main, token: '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')"
+     ]
+    }
+   ],
+   "source": [
+    "from huggingface_hub import HfApi\n",
+    "api = HfApi()\n",
+    "\n",
+    "api.upload_folder(\n",
+    "    folder_path=\"/home/workspace/lechitai/Anti-NSFW/EraX-Anti-NSFW-V1.1\",\n",
+    "    repo_id=\"erax-ai/EraX-NSFW-V1.1\",\n",
+    "    repo_type=\"model\",\n",
+    "    token=\"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\",\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "id": "1aefb2bf-4f5e-4a4f-867e-3401483fd976",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "'/home/workspace/lechitai/Anti-NSFW'"
+      ]
+     },
+     "execution_count": 3,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "pwd"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "bf9486c5-3d8b-4420-bf20-454e427570fd",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.15"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

erax-anti-nsfw-yolo11m-v1.1.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5ec24336b34c2bc8ba7fab6731a260dce55de26e5db87da2b80fd1fea09b1ec8
+size 40495802

erax-anti-nsfw-yolo11n-v1.1.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2df45339e529097f2aaca26fb5a56a0e8f01bafa0ded4cec6cc48ff3aba84eb2
+size 5451432

erax-anti-nsfw-yolo11s-v1.1.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6169d17d11b384f09f47502b71f0c772bb93588df098e8d14ad5d6440e038bfc
+size 19155432