import os
from unstructured_inference.models.base import get_model
def initialize():
"""Download default model or model specified by UNSTRUCTURED_HI_RES_MODEL_NAME environment
variable (avoids subprocesses all doing the same)"""
# If more than one model will be supported and left up to user selection
supported_model = os.environ.get("UNSTRUCTURED_HI_RES_SUPPORTED_MODEL", "")
if supported_model:
for model_name in supported_model.split(","):
get_model(model_name=model_name)
get_model(os.environ.get("UNSTRUCTURED_HI_RES_MODEL_NAME"))