From b794013651b19716ddb9fb025f3cd4bd52a38ab7 Mon Sep 17 00:00:00 2001 From: Taha Douaji Date: Sat, 12 Mar 2022 12:07:32 +0000 Subject: [PATCH] Update README.md --- README.md | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index c8ff72e..1611cdb 100644 --- a/README.md +++ b/README.md @@ -15,13 +15,15 @@ The model was trained on ICDAR2019 Table Dataset ```python from transformers import DetrFeatureExtractor, DetrForObjectDetection from PIL import Image -import requests + image = Image.open("Image path") + feature_extractor = DetrFeatureExtractor.from_pretrained('TahaDouaji/detr-doc-table-detection') model = DetrForObjectDetection.from_pretrained('TahaDouaji/detr-doc-table-detection') + inputs = feature_extractor(images=image, return_tensors="pt") outputs = model(**inputs) -# model predicts bounding boxes and corresponding COCO classes + logits = outputs.logits bboxes = outputs.pred_boxes ``` \ No newline at end of file