Spaces:
Running
Running
Tobias Cornille
commited on
Commit
·
69943f1
1
Parent(s):
dd1ef53
Update index.js
Browse files
index.js
CHANGED
|
@@ -9,6 +9,8 @@ const fileUpload = document.getElementById('upload');
|
|
| 9 |
const imageContainer = document.getElementById('container');
|
| 10 |
const example = document.getElementById('example');
|
| 11 |
const labels = document.getElementById('labels');
|
|
|
|
|
|
|
| 12 |
let parsedLabels = ['human face', 'rocket', 'helmet', 'american flag'];
|
| 13 |
let img = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/astronaut.png';
|
| 14 |
|
|
@@ -33,7 +35,6 @@ fileUpload.addEventListener('change', function (e) {
|
|
| 33 |
// Set up a callback when the file is loaded
|
| 34 |
reader.onload = (e2) => {
|
| 35 |
img = e2.target.result;
|
| 36 |
-
console.log('img in fileUpload event:', img, e2.target.result);
|
| 37 |
detect();
|
| 38 |
}
|
| 39 |
|
|
@@ -43,23 +44,27 @@ fileUpload.addEventListener('change', function (e) {
|
|
| 43 |
|
| 44 |
labels.addEventListener('blur', (e) => {
|
| 45 |
parsedLabels = e.target.value.split(",");
|
| 46 |
-
console.warn(img);
|
| 47 |
if (img) {
|
| 48 |
status.textContent = 'Analysing...';
|
| 49 |
detect();
|
| 50 |
}
|
| 51 |
});
|
| 52 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 53 |
// Detect objects in the image
|
| 54 |
async function detect() {
|
| 55 |
-
console.warn(parsedLabels);
|
| 56 |
imageContainer.innerHTML = '';
|
| 57 |
imageContainer.style.backgroundImage = `url(${img})`;
|
| 58 |
|
| 59 |
status.textContent = 'Analysing...';
|
| 60 |
|
| 61 |
const output = await detector(img, parsedLabels, { percentage: true });
|
| 62 |
-
console.warn(output);
|
| 63 |
status.textContent = '';
|
| 64 |
output.forEach(renderBox);
|
| 65 |
|
|
|
|
| 9 |
const imageContainer = document.getElementById('container');
|
| 10 |
const example = document.getElementById('example');
|
| 11 |
const labels = document.getElementById('labels');
|
| 12 |
+
const form = document.getElementById('form');
|
| 13 |
+
|
| 14 |
let parsedLabels = ['human face', 'rocket', 'helmet', 'american flag'];
|
| 15 |
let img = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/astronaut.png';
|
| 16 |
|
|
|
|
| 35 |
// Set up a callback when the file is loaded
|
| 36 |
reader.onload = (e2) => {
|
| 37 |
img = e2.target.result;
|
|
|
|
| 38 |
detect();
|
| 39 |
}
|
| 40 |
|
|
|
|
| 44 |
|
| 45 |
labels.addEventListener('blur', (e) => {
|
| 46 |
parsedLabels = e.target.value.split(",");
|
|
|
|
| 47 |
if (img) {
|
| 48 |
status.textContent = 'Analysing...';
|
| 49 |
detect();
|
| 50 |
}
|
| 51 |
});
|
| 52 |
|
| 53 |
+
form.addEventListener('submit', () => {
|
| 54 |
+
if (img) {
|
| 55 |
+
status.textContent = 'Analysing...';
|
| 56 |
+
detect();
|
| 57 |
+
}
|
| 58 |
+
})
|
| 59 |
+
|
| 60 |
// Detect objects in the image
|
| 61 |
async function detect() {
|
|
|
|
| 62 |
imageContainer.innerHTML = '';
|
| 63 |
imageContainer.style.backgroundImage = `url(${img})`;
|
| 64 |
|
| 65 |
status.textContent = 'Analysing...';
|
| 66 |
|
| 67 |
const output = await detector(img, parsedLabels, { percentage: true });
|
|
|
|
| 68 |
status.textContent = '';
|
| 69 |
output.forEach(renderBox);
|
| 70 |
|