[SuperPoint] Fix doc example (#29816)

[SuperPoint] Fix doc example
huggingface · Mar 22, 2024 · c5f0288 · c5f0288
1 parent 7e1413d
commit c5f0288
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/transformers/models/superpoint/modeling_superpoint.py b/src/transformers/models/superpoint/modeling_superpoint.py
@@ -423,7 +423,7 @@ def forward(
         Examples:
 
         ```python
-        >>> from transformers import AutoImageProcessor, AutoModel
+        >>> from transformers import AutoImageProcessor, SuperPointForKeypointDetection
         >>> import torch
         >>> from PIL import Image
         >>> import requests
@@ -432,7 +432,7 @@ def forward(
         >>> image = Image.open(requests.get(url, stream=True).raw)
 
         >>> processor = AutoImageProcessor.from_pretrained("magic-leap-community/superpoint")
-        >>> model = AutoModel.from_pretrained("magic-leap-community/superpoint")
+        >>> model = SuperPointForKeypointDetection.from_pretrained("magic-leap-community/superpoint")
 
         >>> inputs = processor(image, return_tensors="pt")
         >>> outputs = model(**inputs)