Update README.md
Browse files
README.md
CHANGED
|
@@ -61,6 +61,8 @@ model = DetrForSegmentation.from_pretrained(
|
|
| 61 |
"cmarkea/detr-layout-detection"
|
| 62 |
)
|
| 63 |
|
|
|
|
|
|
|
| 64 |
with torch.inference_mode():
|
| 65 |
input_ids = img_proc(img, return_tensors='pt')
|
| 66 |
output = model(**input_ids)
|
|
|
|
| 61 |
"cmarkea/detr-layout-detection"
|
| 62 |
)
|
| 63 |
|
| 64 |
+
img: PIL.Image
|
| 65 |
+
|
| 66 |
with torch.inference_mode():
|
| 67 |
input_ids = img_proc(img, return_tensors='pt')
|
| 68 |
output = model(**input_ids)
|