Update docstrings

qimaqi · Aug 6, 2024 · 43c385c · 43c385c
1 parent 322aa3e
commit 43c385c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sam2/sam2_image_predictor.py b/sam2/sam2_image_predictor.py
@@ -65,7 +65,7 @@ def __init__(
     @classmethod
     def from_pretrained(cls, model_id: str, **kwargs) -> "SAM2ImagePredictor":
         """
-        Load a pretrained model from the Hugging Face model hub.
+        Load a pretrained model from the Hugging Face hub.
 
         Arguments:
           model_id (str): The Hugging Face repository ID.

diff --git a/sam2/sam2_video_predictor.py b/sam2/sam2_video_predictor.py
@@ -106,7 +106,7 @@ def init_state(
     @classmethod
     def from_pretrained(cls, model_id: str, **kwargs) -> "SAM2VideoPredictor":
         """
-        Load a pretrained model from the Hugging Face model hub.
+        Load a pretrained model from the Hugging Face hub.
 
         Arguments:
           model_id (str): The Hugging Face repository ID.