from __future__ import annotations

class SegmentationModel:
    @classmethod
    def from_pretrained(
        cls,
        model: str,
        *,
        use_hf_token: str | bool | None = True,
    ) -> SegmentationModel: ...


class EmbeddingModel:
    @classmethod
    def from_pretrained(
        cls,
        model: str,
        *,
        use_hf_token: str | bool | None = True,
    ) -> EmbeddingModel: ...
