Update README

README.md

@@ -12,7 +12,7 @@ pipeline_tag: depth-estimation
 
 # TIPSv2 — g/14 DPT Heads
 
-DPT (Dense Prediction Transformer) heads for depth estimation, surface normal prediction, and semantic segmentation (ADE20K, 150 classes) on top of the [TIPSv2 g/14](https://huggingface.co/google/tipsv2-g14) backbone. The backbone is loaded automatically.
+DPT (Dense Prediction Transformer) heads for depth estimation, surface normal prediction, and semantic segmentation on top of the frozen [TIPSv2 g/14](https://huggingface.co/google/tipsv2-g14) backbone. The backbone is loaded automatically. The depth and normals heads are trained on the NYU Depth V2 dataset and segmentation is trained on the ADE20K dataset (150 classes).
 
 ## Usage
 
@@ -47,6 +47,7 @@ seg = model.predict_segmentation(pixel_values)
 
 - **Backbone**: [TIPSv2 g/14](google/tipsv2-g14) (loaded automatically)
 - **Heads**: ~185M total params (depth + normals + segmentation)
+- **Depth & normals**: NYU Depth V2
 - **Segmentation**: ADE20K, 150 classes
 - **Input**: images in `[0, 1]` range, any resolution (multiples of 14 recommended)