Unverified Commit 39e76d76 authored by NielsRogge's avatar NielsRogge Committed by GitHub
Browse files

Fix doc tests (#18397)


Co-authored-by: default avatarNiels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
parent 11413711
...@@ -1419,7 +1419,7 @@ class DetrForObjectDetection(DetrPreTrainedModel): ...@@ -1419,7 +1419,7 @@ class DetrForObjectDetection(DetrPreTrainedModel):
... if score > 0.9: ... if score > 0.9:
... print( ... print(
... f"Detected {model.config.id2label[label.item()]} with confidence " ... 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}"
... ) ... )
Detected remote with confidence 0.998 at location [40.16, 70.81, 175.55, 117.98] Detected remote with confidence 0.998 at location [40.16, 70.81, 175.55, 117.98]
Detected remote with confidence 0.996 at location [333.24, 72.55, 368.33, 187.66] Detected remote with confidence 0.996 at location [333.24, 72.55, 368.33, 187.66]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment