Xenova HF staff commited on
Commit
74a6723
1 Parent(s): 7b6fd56

Update index.js

Browse files
Files changed (1) hide show
  1. index.js +2 -2
index.js CHANGED
@@ -16,7 +16,7 @@ const THRESHOLD = 0.25;
16
  status.textContent = 'Loading model...';
17
  const model_id = 'onnx-community/yolov10s';
18
  const processor = await AutoProcessor.from_pretrained(model_id);
19
- const model = await AutoModel.from_pretrained(model_id, { quantized: false });
20
  status.textContent = 'Ready';
21
 
22
  example.addEventListener('click', (e) => {
@@ -88,7 +88,7 @@ function renderBox([xmin, ymin, xmax, ymax, score, id], [w, h]) {
88
 
89
  // Draw label
90
  const labelElement = document.createElement('span');
91
- labelElement.textContent = model.config.id2label[id];
92
  labelElement.className = 'bounding-box-label';
93
  labelElement.style.backgroundColor = color;
94
 
 
16
  status.textContent = 'Loading model...';
17
  const model_id = 'onnx-community/yolov10s';
18
  const processor = await AutoProcessor.from_pretrained(model_id);
19
+ const model = await AutoModel.from_pretrained(model_id);
20
  status.textContent = 'Ready';
21
 
22
  example.addEventListener('click', (e) => {
 
88
 
89
  // Draw label
90
  const labelElement = document.createElement('span');
91
+ labelElement.textContent = `${model.config.id2label[id]} (${score.toFixed(2)}%)`;
92
  labelElement.className = 'bounding-box-label';
93
  labelElement.style.backgroundColor = color;
94