hubconf.py

# autopep8: off
dependencies = ["torch", "torchvision"]

import sys
from pathlib import Path

repo_dir = Path(__file__).parent
multimodal_src_dir = repo_dir / "hi-ml-multimodal" / "src" / "health_multimodal" / "image"
sys.path.append(str(multimodal_src_dir))

from model import ImageModel
from model import get_biovil_image_encoder as _biovil_resnet

# autopep8: on


def biovil_resnet(pretrained: bool = False) -> ImageModel:
    """Get BioViL image encoder.

    :param pretrained: Load pretrained weights from a checkpoint.
    """
    model = _biovil_resnet(pretrained=pretrained)
    return model