Spaces:
Running
Running
Tobias Cornille
commited on
Commit
·
500ed37
1
Parent(s):
ed29a46
Update index.js
Browse files
index.js
CHANGED
|
@@ -9,7 +9,7 @@ const fileUpload = document.getElementById('upload');
|
|
| 9 |
const imageContainer = document.getElementById('container');
|
| 10 |
const example = document.getElementById('example');
|
| 11 |
|
| 12 |
-
const EXAMPLE_URL = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/
|
| 13 |
|
| 14 |
// Create a new object detection pipeline
|
| 15 |
status.textContent = 'Loading model...';
|
|
@@ -42,7 +42,8 @@ async function detect(img) {
|
|
| 42 |
imageContainer.style.backgroundImage = `url(${img})`;
|
| 43 |
|
| 44 |
status.textContent = 'Analysing...';
|
| 45 |
-
const
|
|
|
|
| 46 |
status.textContent = '';
|
| 47 |
output.forEach(renderBox);
|
| 48 |
}
|
|
|
|
| 9 |
const imageContainer = document.getElementById('container');
|
| 10 |
const example = document.getElementById('example');
|
| 11 |
|
| 12 |
+
const EXAMPLE_URL = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/astronaut.png';
|
| 13 |
|
| 14 |
// Create a new object detection pipeline
|
| 15 |
status.textContent = 'Loading model...';
|
|
|
|
| 42 |
imageContainer.style.backgroundImage = `url(${img})`;
|
| 43 |
|
| 44 |
status.textContent = 'Analysing...';
|
| 45 |
+
const candidate_labels = ['human face', 'rocket', 'helmet', 'american flag'];
|
| 46 |
+
const output = await detector('https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/astronaut.png', candidate_labels);
|
| 47 |
status.textContent = '';
|
| 48 |
output.forEach(renderBox);
|
| 49 |
}
|