Update README.md

README.md

@@ -55,7 +55,7 @@ MatchAnything-ELOFTR is designed for a vast array of applications requiring robu
 
 Here is a quick example of using the model for matching a pair of images.
 ```python
-from transformers import AutoImageProcessor, AutoModel
+from transformers import AutoImageProcessor, AutoModelForKeypointMatching
 from transformers.image_utils import load_image
 import torch
 
@@ -67,7 +67,7 @@ images = [image1, image2]
 
 # Load the processor and model from the Hugging Face Hub
 processor = AutoImageProcessor.from_pretrained("zju-community/matchanything_eloftr")
-model = AutoModel.from_pretrained("zju-community/matchanything_eloftr")
+model = AutoModelForKeypointMatching.from_pretrained("zju-community/matchanything_eloftr")
 
 # Process images and get model outputs
 inputs = processor(images, return_tensors="pt")