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 |
|