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