Tobias Cornille commited on
Commit
6296a62
·
1 Parent(s): fd01f49

Update index.js

Browse files
Files changed (1) hide show
  1. index.js +12 -7
index.js CHANGED
@@ -9,7 +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 candidate_labels = ['human face', 'rocket', 'helmet', 'american flag'];
 
13
 
14
  const EXAMPLE_URL = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/astronaut.png';
15
 
@@ -32,22 +33,26 @@ fileUpload.addEventListener('change', function (e) {
32
 
33
 
34
  // Set up a callback when the file is loaded
35
- reader.onload = e2 => detect(e2.target.result);
 
 
 
36
 
37
  reader.readAsDataURL(file);
38
  });
39
 
40
 
41
  labels.addEventListener('blur', (e) => {
42
- candidate_labels = e.target.value.split(",");
43
- console.warn(candidate_labels, reader);
44
- if (reader.readyState === "DONE") {
45
- detect(reader.result);
 
46
  }
47
  });
48
 
49
  // Detect objects in the image
50
- async function detect(img) {
51
  imageContainer.innerHTML = '';
52
  imageContainer.style.backgroundImage = `url(${img})`;
53
 
 
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;
14
 
15
  const EXAMPLE_URL = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/astronaut.png';
16
 
 
33
 
34
 
35
  // Set up a callback when the file is loaded
36
+ reader.onload = e2 => {
37
+ img = e2.target.result;
38
+ detect(img, parsedLabels);
39
+ }
40
 
41
  reader.readAsDataURL(file);
42
  });
43
 
44
 
45
  labels.addEventListener('blur', (e) => {
46
+ parsedLabels = e.target.value.split(",");
47
+ console.warn(parsedLabels, reader);
48
+ if (img) {
49
+ status.textContent = 'Analysing...';
50
+ detect(img, parsedLabels);
51
  }
52
  });
53
 
54
  // Detect objects in the image
55
+ async function detect(img, candidate_labels) {
56
  imageContainer.innerHTML = '';
57
  imageContainer.style.backgroundImage = `url(${img})`;
58