Update README.md

README.md

@@ -52,12 +52,12 @@ Now, let's compare the performance of this model with other models.
 
 ```python
 import torch
-from transformers import AutoImageProcessor, AutoModel
+from transformers import AutoImageProcessor, BeitForSemanticSegmentation
 
 img_proc = AutoImageProcessor.from_pretrained(
     "cmarkea/dit-base-layout-detection"
 )
-model = AutoModel.from_pretrained(
+model = BeitForSemanticSegmentation.from_pretrained(
     "cmarkea/dit-base-layout-detection"
 )