Update README.md

README.md

@@ -19,6 +19,7 @@ model-index:
       value: 0.0937
       name: Loss
 license: apache-2.0
+library_name: transformers.js
 ---
 
 # vit-base-nsfw-detector
@@ -66,6 +67,41 @@ print("Predicted class:", model.config.id2label[predicted_class_idx])
 # Predicted class: sfw
 ```
 
+Usage with Transformers.js
+```js
+import { pipeline, env } from 'https://cdn.jsdelivr.net/npm/@xenova/[email protected]';
+
+// Load the image classification model
+const classifier = await pipeline('image-classification', 'AdamCodd/vit-base-nsfw-detector');
+
+// Function to fetch and classify an image from a URL
+async function classifyImage(url) {
+  try {
+    const response = await fetch(url);
+    if (!response.ok) throw new Error('Failed to load image');
+
+    const blob = await response.blob();
+    const image = new Image();
+    const imagePromise = new Promise((resolve, reject) => {
+      image.onload = () => resolve(image);
+      image.onerror = reject;
+      image.src = URL.createObjectURL(blob);
+    });
+
+    const img = await imagePromise; // Ensure the image is loaded
+    const classificationResults = await classifier([img.src]); // Classify the image
+    console.log('Predicted class: ', classificationResults[0].label);
+  } catch (error) {
+    console.error('Error classifying image:', error);
+  }
+}
+
+// Example usage
+classifyImage('http://images.cocodataset.org/val2017/000000039769.jpg');
+// Predicted class: sfw
+```
+
+
 The model has been trained on a variety of images (realistic, 3D, drawings), yet it is not perfect and some images may be wrongly classified as NSFW when they are not.
 
 ## Training and evaluation data