Add print statements with descriptive comments to code examples

README.md

@@ -36,14 +36,15 @@ pixel_values = transform(image).unsqueeze(0).cuda()
 
 # All tasks at once
 outputs = model(pixel_values)
-outputs.depth          # (B, 1, H, W)
-outputs.normals        # (B, 3, H, W)
-outputs.segmentation   # (B, 150, H, W)
+print(outputs.depth.shape)         # (1, 1, 448, 448) — depth map
+print(outputs.normals.shape)       # (1, 3, 448, 448) — surface normals
+print(outputs.segmentation.shape)  # (1, 150, 448, 448) — segmentation logits
 
 # Or individual tasks (only runs the requested head)
 depth = model.predict_depth(pixel_values)
 normals = model.predict_normals(pixel_values)
 seg = model.predict_segmentation(pixel_values)
+print(seg.argmax(dim=1).shape)     # (1, 448, 448) — per-pixel class prediction
 ```
 
 ## Model details