worldcoin / open-iris

Open Iris Recognition Inference System (IRIS)
MIT License
222 stars 42 forks source link

Model loading constructors: via HF download and specifying model path. #24

Closed wiktorlazarski closed 3 months ago

wiktorlazarski commented 3 months ago

Model loading constructors: via HF download and specifying model path.

PR description

Introduce a possibility to create iris.MultilabelSegmentation based on model name available in MultilabelSemanticSegmentationInterface.HUGGING_FACE_REPO_ID = "Worldcoin/iris-semantic-segmentation" repo or by specifying the path to the model weights.

Type

Checklist