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