DHEIVER commited on
Commit
fa1d439
·
verified ·
1 Parent(s): 57060db

Update index.js

Browse files
Files changed (1) hide show
  1. index.js +26 -74
index.js CHANGED
@@ -1,79 +1,31 @@
1
- import { pipeline, env } from 'https://cdn.jsdelivr.net/npm/@xenova/transformers@2.10.1';
2
-
3
- // Since we will download the model from the Hugging Face Hub, we can skip the local model check
4
- env.allowLocalModels = false;
5
-
6
- // Reference the elements that we will need
7
- const status = document.getElementById('status');
8
- 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/city-streets.jpg';
13
-
14
- // Create a new object detection pipeline
15
- status.textContent = 'Loading model...';
16
- const detector = await pipeline('object-detection', 'Xenova/detr-resnet-50');
17
- status.textContent = 'Ready';
18
-
19
- example.addEventListener('click', (e) => {
20
- e.preventDefault();
21
- detect(EXAMPLE_URL);
22
- });
23
-
24
- fileUpload.addEventListener('change', function (e) {
25
- const file = e.target.files[0];
26
- if (!file) {
27
- return;
28
  }
29
 
30
- const reader = new FileReader();
 
 
 
 
 
 
 
31
 
32
- // Set up a callback when the file is loaded
33
- reader.onload = e2 => detect(e2.target.result);
34
 
35
- reader.readAsDataURL(file);
 
36
  });
37
-
38
-
39
- // Detect objects in the image
40
- async function detect(img) {
41
- imageContainer.innerHTML = '';
42
- imageContainer.style.backgroundImage = `url(${img})`;
43
-
44
- status.textContent = 'Analysing...';
45
- const output = await detector(img, {
46
- threshold: 0.5,
47
- percentage: true,
48
- });
49
- status.textContent = '';
50
- output.forEach(renderBox);
51
- }
52
-
53
- // Render a bounding box and label on the image
54
- function renderBox({ box, label }) {
55
- const { xmax, xmin, ymax, ymin } = box;
56
-
57
- // Generate a random color for the box
58
- const color = '#' + Math.floor(Math.random() * 0xFFFFFF).toString(16).padStart(6, 0);
59
-
60
- // Draw the box
61
- const boxElement = document.createElement('div');
62
- boxElement.className = 'bounding-box';
63
- Object.assign(boxElement.style, {
64
- borderColor: color,
65
- left: 100 * xmin + '%',
66
- top: 100 * ymin + '%',
67
- width: 100 * (xmax - xmin) + '%',
68
- height: 100 * (ymax - ymin) + '%',
69
- })
70
-
71
- // Draw label
72
- const labelElement = document.createElement('span');
73
- labelElement.textContent = label;
74
- labelElement.className = 'bounding-box-label';
75
- labelElement.style.backgroundColor = color;
76
-
77
- boxElement.appendChild(labelElement);
78
- imageContainer.appendChild(boxElement);
79
- }
 
1
+ // app.js
2
+
3
+ document.addEventListener('DOMContentLoaded', function () {
4
+ const statusElement = document.getElementById('status');
5
+ const outputElement = document.getElementById('output');
6
+ const commandButton = document.getElementById('commandButton');
7
+
8
+ // Função para simular uma atualização de status
9
+ function updateStatus() {
10
+ fetch('status.json')
11
+ .then(response => response.json())
12
+ .then(data => {
13
+ statusElement.textContent = `Status: ${data.status}`;
14
+ });
 
 
 
 
 
 
 
 
 
 
 
 
 
15
  }
16
 
17
+ // Função para simular uma resposta de comando
18
+ function sendCommand() {
19
+ fetch('command.json')
20
+ .then(response => response.json())
21
+ .then(data => {
22
+ outputElement.textContent = `Comando recebido: ${data.command}`;
23
+ });
24
+ }
25
 
26
+ // Atualizar o status quando a página carrega
27
+ updateStatus();
28
 
29
+ // Adicionar evento ao botão
30
+ commandButton.addEventListener('click', sendCommand);
31
  });