From 05f4697acec2da61b2133576e3089cb1993d8ba4 Mon Sep 17 00:00:00 2001
From: GamerYuan <f2pplayer0801@gmail.com>
Date: Mon, 15 Jun 2026 20:25:34 +0800
Subject: [PATCH] chore: use new nsfw classifier model

---
 Models/nsfw-classifier/config.json            | 37 ++++++++++---------
 .../nsfw-classifier/freepik-nsfw-model.onnx   |  3 ++
 Models/nsfw-classifier/model.onnx             |  3 --
 3 files changed, 23 insertions(+), 20 deletions(-)
 create mode 100644 Models/nsfw-classifier/freepik-nsfw-model.onnx
 delete mode 100644 Models/nsfw-classifier/model.onnx

diff --git a/Models/nsfw-classifier/config.json b/Models/nsfw-classifier/config.json
index 9e7f94a..4edadb7 100644
--- a/Models/nsfw-classifier/config.json
+++ b/Models/nsfw-classifier/config.json
@@ -1,36 +1,39 @@
 {
-  "architecture": "vit_tiny_patch16_384",
+  "architecture": "eva02_base_patch14_448",
   "classifier": "head",
-  "crop_mode": "center",
+  "crop_mode": "squash",
   "crop_pct": 1.0,
-  "custom_load": true,
-  "export_model_type": "vit_tiny_patch16_384",
+  "custom_load": false,
+  "export_model_type": "eva02_base_patch14_448",
   "first_conv": "patch_embed.proj",
   "fixed_input_size": true,
-  "hf_hub_id": "Marqo/nsfw-image-detection-384",
+  "hf_hub_id": "Freepik/nsfw_image_detector",
   "input_size": [
     3,
-    384,
-    384
+    448,
+    448
   ],
   "interpolation": "bicubic",
   "label_names": [
-    "NSFW",
-    "SFW"
+    "neutral",
+    "low",
+    "medium",
+    "high"
   ],
+  "license": "mit",
   "mean": [
-    0.5,
-    0.5,
-    0.5
+    0.48145466,
+    0.4578275,
+    0.40821073
   ],
-  "num_classes": 2,
+  "num_classes": 4,
   "pool_size": null,
   "source": "hf-hub",
   "std": [
-    0.5,
-    0.5,
-    0.5
+    0.26862954,
+    0.26130258,
+    0.27577711
   ],
-  "tag": "augreg_in21k_ft_in1k",
+  "tag": "mim_in22k_ft_in22k_in1k",
   "transformers_version": "4.51.3"
 }
diff --git a/Models/nsfw-classifier/freepik-nsfw-model.onnx b/Models/nsfw-classifier/freepik-nsfw-model.onnx
new file mode 100644
index 0000000..034278c
--- /dev/null
+++ b/Models/nsfw-classifier/freepik-nsfw-model.onnx
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ad08dd0f69be614ece6bee0b9aa8b845cc1a981b353ffb17862fa9efbdd6d9f8
+size 102311827
diff --git a/Models/nsfw-classifier/model.onnx b/Models/nsfw-classifier/model.onnx
deleted file mode 100644
index e59a30e..0000000
--- a/Models/nsfw-classifier/model.onnx
+++ /dev/null
@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7b81155313d894bba3a3b9bace059a6da2a0c509d10cea43fcb0b3c5a4edd26d
-size 22527617