Update README.md
Browse files
README.md
CHANGED
@@ -55,10 +55,10 @@ from transformers import AutoImageProcessor
|
|
55 |
from transformers.models.detr import DetrForSegmentation
|
56 |
|
57 |
img_proc = AutoImageProcessor.from_pretrained(
|
58 |
-
"
|
59 |
)
|
60 |
model = DetrForSegmentation.from_pretrained(
|
61 |
-
"
|
62 |
)
|
63 |
|
64 |
with torch.inference_mode():
|
|
|
55 |
from transformers.models.detr import DetrForSegmentation
|
56 |
|
57 |
img_proc = AutoImageProcessor.from_pretrained(
|
58 |
+
"cmarkea/detr-layout-detection"
|
59 |
)
|
60 |
model = DetrForSegmentation.from_pretrained(
|
61 |
+
"cmarkea/detr-layout-detection"
|
62 |
)
|
63 |
|
64 |
with torch.inference_mode():
|