Update README.md

README.md

@@ -1,5 +1,5 @@
 ---
-license: gpl-3.0
+license: apache-2.0
 ---
 
 <div align="center">
@@ -19,11 +19,11 @@ Custom trained models for the soon to be released **Forbidden Vision** ComfyUI c
 
 ## 📦 Models
 
-All models were trained on a custom dataset of **7k+ images** (a mix of SFW/NSFW from Civitai and my own curated images).  
+All models were trained on a custom dataset of **10k+ images** (a mix of SFW/NSFW from Civitai, Danbooru, real photos and my own curated images).  
 The goal was to cover style diversity across anime and realistic domains.
 
-### 🟢 Face Detection (YOLO fine-tunes)
-- **Base:** YOLO face detection models (GPL-3.0).  
+### 🟢 Face Detection (YOLO 11 small)
+- **Base:** YOLO face detection models   
 - **Variants:**  
   - `ForbiddenVision_facedetect_yolov11s.pt` → YOLO Small fine-tune -> generally more robust and reliable
   - `ForbiddenVision_facedetect_yolov11n.pt` → YOLO Nano fine-tune -> if you need maximum speed and lower vram usage
@@ -33,19 +33,13 @@ The goal was to cover style diversity across anime and realistic domains.
 ---
 
 ### 🔵 Face Segmentation (EfficientNet-B0 advprop)
-- **Base:** timm EfficientNet-B0 (advprop, Apache-2.0).  
+- **Base:** timm EfficientNet-B0   
 - **Training:** Besides manually annotated civitai and custom images, the model used an equal split of <a href="https://github.com/switchablenorms/CelebAMask-HQ">CelebAMask-HQ</a>
 - **Usage:** Provides clean masks for downstream processing.  
 
 ---
 
 ### 🟣 Landmark Rotation (MobileNetV3-Small)
-- **Base:** MobileNetV3-Small (Apache-2.0).  
+- **Base:** MobileNetV3-Small 
 - **Training:** Trained for landmark-based rotation correction.  
-- **Usage:** Aligns faces to level eyes for consistent preprocessing.  
-
----
-
-⚖️ **Licensing Note**  
-This repository is distributed under **GPL-3.0** due to the YOLO fine-tunes.  
-The segmentation and rotation models are originally Apache-2.0 based, but included here under GPL for consistency.
+- **Usage:** Aligns faces to level eyes for consistent preprocessing.