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)
|