Update README.md

README.md

@@ -49,8 +49,35 @@ This is an ONNX version of [abhilash88/age-gender-prediction](https://huggingfac
 ## Usage with Transformers.js
 
 
-See the pipeline documentation for `image-classification`: https://huggingface.co/docs/transformers.js/api/pipelines#module_pipelines.ImageClassificationPipeline
-
+```js
+import { AutoModel, AutoProcessor, load_image } from "@huggingface/transformers";
+
+// Load model and processor
+const model_id = 'onnx-community/age-gender-prediction-ONNX';
+const model = await AutoModel.from_pretrained(model_id);
+const processor = await AutoProcessor.from_pretrained(model_id);
+
+// Load and preprocess image
+const image = await load_image('https://images.pexels.com/photos/614810/pexels-photo-614810.jpeg?w=720');
+const inputs = await processor(image);
+
+// Run model
+const { logits } = await model(inputs);
+
+// Extract age and gender
+const [ageLogits, genderLogits] = logits.tolist()[0];
+const age = Math.min(Math.max(Math.round(ageLogits), 0), 100);
+const gender = genderLogits >= 0.5 ? 'Female' : 'Male';
+const confidence = Math.max(genderLogits, 1 - genderLogits);
+console.table({ age, gender, confidence });
+// ┌────────────┬────────────────────┐
+// │ (index)    │ Values             │
+// ├────────────┼────────────────────┤
+// │ age        │ 26                 │
+// │ gender     │ 'Male'             │
+// │ confidence │ 0.9973341226577759 │
+// └────────────┴────────────────────┘
+```
 
 ---