Update README.md
This commit is contained in:
parent
8c174de1db
commit
c783425425
|
@ -61,7 +61,7 @@ for score, label, box in zip(results["scores"], results["labels"], results["boxe
|
|||
if score > 0.9:
|
||||
print(
|
||||
f"Detected {model.config.id2label[label.item()]} with confidence "
|
||||
f"{round(score.item(), 3)} at location {box}."
|
||||
f"{round(score.item(), 3)} at location {box}"
|
||||
)
|
||||
```
|
||||
This should output:
|
||||
|
|
Loading…
Reference in New Issue