Spaces:
Running
Running
Update app.py
Browse files
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,
|
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
|