marie-ai
marie-ai copied to clipboard
Add support for loading models from google drive
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)