cmarkea
/

dit-base-layout-detection

Image Segmentation

Inference Endpoints

Model card Files Files and versions Community

Cyrile commited on Sep 11

Commit

9847b1d

•

1 Parent(s): 5116a9d

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

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

 ```python
 import torch
+from transformers import AutoImageProcessor, BeitForSemanticSegmentation
 img_proc = AutoImageProcessor.from_pretrained(
     "cmarkea/dit-base-layout-detection"
 )
+model = BeitForSemanticSegmentation.from_pretrained(
     "cmarkea/dit-base-layout-detection"
 )