Update README.md
Browse filesChanged model name
README.md
CHANGED
@@ -50,8 +50,8 @@ import requests
|
|
50 |
url = "http://images.cocodataset.org/val2017/000000039769.jpg"
|
51 |
image = Image.open(requests.get(url, stream=True).raw)
|
52 |
|
53 |
-
processor = AutoImageProcessor.from_pretrained("Aryn/deformable-detr")
|
54 |
-
model = DeformableDetrForObjectDetection.from_pretrained("Aryn/deformable-detr")
|
55 |
|
56 |
inputs = processor(images=image, return_tensors="pt")
|
57 |
outputs = model(**inputs)
|
|
|
50 |
url = "http://images.cocodataset.org/val2017/000000039769.jpg"
|
51 |
image = Image.open(requests.get(url, stream=True).raw)
|
52 |
|
53 |
+
processor = AutoImageProcessor.from_pretrained("Aryn/deformable-detr-DocLayNet")
|
54 |
+
model = DeformableDetrForObjectDetection.from_pretrained("Aryn/deformable-detr-DocLayNet")
|
55 |
|
56 |
inputs = processor(images=image, return_tensors="pt")
|
57 |
outputs = model(**inputs)
|