From d60c294e8574660ac2c5cbb2239ce9dde6f41f2b Mon Sep 17 00:00:00 2001
From: Niels Rogge <niels.rogge1@gmail.com>
Date: Fri, 5 Aug 2022 13:31:07 +0000
Subject: [PATCH] Improve code snippet

---
 README.md | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 3b184cd..7069046 100644
--- a/README.md
+++ b/README.md
@@ -24,6 +24,7 @@ model = DetrForObjectDetection.from_pretrained('TahaDouaji/detr-doc-table-detect
 inputs = feature_extractor(images=image, return_tensors="pt")
 outputs = model(**inputs)
 
-logits = outputs.logits
-bboxes = outputs.pred_boxes
+# convert outputs (bounding boxes and class logits) to COCO API
+target_sizes = torch.tensor([image.size[::-1]])
+results = feature_extractor.post_process(outputs, target_sizes=target_sizes)[0]
 ```
\ No newline at end of file