diff --git a/sam2/sam2_image_predictor.py b/sam2/sam2_image_predictor.py index 6fde883..f6f9a5a 100644 --- a/sam2/sam2_image_predictor.py +++ b/sam2/sam2_image_predictor.py @@ -65,7 +65,7 @@ class SAM2ImagePredictor: @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 index 6e27efb..fe8702b 100644 --- a/sam2/sam2_video_predictor.py +++ b/sam2/sam2_video_predictor.py @@ -106,7 +106,7 @@ class SAM2VideoPredictor(SAM2Base): @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.