JosephTK commited on
Commit
97f1b93
·
1 Parent(s): adca989

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def detect(image):
29
  counts = {}
30
 
31
  for score, label, box in zip(results["scores"], results["labels"], results["boxes"]):
32
- box = [round(i, 2) for i in box.tolist()]
33
  label_name = model.config.id2label[label.item()]
34
  if label_name not in counts:
35
  counts[label_name] = 0
 
29
  counts = {}
30
 
31
  for score, label, box in zip(results["scores"], results["labels"], results["boxes"]):
32
+ box = [round(i, 4) for i in box.tolist()]
33
  label_name = model.config.id2label[label.item()]
34
  if label_name not in counts:
35
  counts[label_name] = 0