search

marieai / marie-ai

Integrate AI-powered Document Analysis Pipelines
MIT License
61 stars 5 forks source link

Add support for loading models from google drive #33

Open gregbugaj opened 1 year ago

gregbugaj commented 1 year ago

Add support for loading models from google drive in the ModelRegistry

Possible syntax.

    ModelRegistry.register_provider("gdrive")

    _name_or_path = "rms/layoutlmv3-large-corr-ner"
    kwargs = {
                      "provider":"gdrive",
                       "__model_path__": __model_path__
    }
    _name_or_path = ModelRegistry.get_local_path(_name_or_path, **kwargs)