Tobias Cornille commited on
Commit
69943f1
·
1 Parent(s): dd1ef53

Update index.js

Browse files
Files changed (1) hide show
  1. index.js +9 -4
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