Update index.js

index.js

@@ -13,7 +13,7 @@ const EXAMPLE_URL = 'https://huggingface.co/datasets/Xenova/transformers.js-docs
 
 // Create a new object detection pipeline
 status.textContent = 'Loading model...';
-const detector = await pipeline('zero-shot-object-detection', 'Xenova/owlvit-base-patch32');
+const detector = await pipeline('zero-shot-object-detection', 'Xenova/owlvit-base-patch32', {percentage: true});
 status.textContent = 'Ready';
 
 example.addEventListener('click', (e) => {
@@ -42,17 +42,11 @@ async function detect(img) {
     imageContainer.style.backgroundImage = `url(${img})`;
 
     status.textContent = 'Analysing...';
-    
-    const image = new Image();
-    image.src = img;
   
     const candidate_labels = ['human face', 'rocket', 'helmet', 'american flag'];
-    image.onload = async function () {
-      const output = await detector(img, candidate_labels);
-      const relativeOutput = output.map(({box, label}) => ({box: {xmax: box.xmax / image.width, xmin: box.xmin / image.width, ymax: box.ymax / image.height, ymin: box.ymin / image.height}, label}));
-      status.textContent = '';
-      relativeOutput.forEach(renderBox);
-    };
+    const output = await detector(img, candidate_labels);
+    status.textContent = '';
+    relativeOutput.forEach(renderBox);
    
 }