Update README.md

README.md

@@ -31,10 +31,10 @@ This is the model card of a 🤗 transformers model that has been pushed on the
 from transformers.models.detr import DetrForSegmentation
 
 img_proc = AutoImageProcessor.from_pretrained(
-    "ArkeaIAF/detr-layout-detection"
+    "ArkeaIAF/detr-base-layout-detection"
 )
 model = DetrForSegmentation.from_pretrained(
-    "ArkeaIAF/detr-layout-detection"
+    "ArkeaIAF/detr-base-layout-detection"
 )
 
 with torch.inference_mode():
@@ -61,7 +61,7 @@ bbox_pred = img_proc.post_process_object_detection(
 ```
 @online{DeDetrLay,
   AUTHOR = {Cyrile Delestre},
-  URL = {https://huggingface.co/cmarkea/detr-layout-detection},
+  URL = {https://huggingface.co/cmarkea/detr-base-layout-detection},
   YEAR = {2024},
   KEYWORDS = {Image Processing ; Transformers ; Layout},
 }