Update README.md

README.md

@@ -26,10 +26,10 @@ Users should be mindful of the model’s limitations, particularly in handling d
 
 How to Get Started with the Model
 ``` python
-from transformers import AutoTokenizer, AutoModelForTokenClassification
+from transformers import AutoTokenizer, AutoModel
 
 tokenizer = AutoTokenizer.from_pretrained("tobiadefami/docmodel-base")
-model = AutoModelForTokenClassification.from_pretrained("tobiadefami/docmodel-base")
+model = AutoModel.from_pretrained("tobiadefami/docmodel-base")
 
 # Example usage
 inputs = tokenizer("Your document text here...", return_tensors="pt")