logic.py

from transformers import UdopProcessor, UdopForConditionalGeneration


def load_Model(): 
    repo_id = "microsoft/udop-large"
    processor = UdopProcessor.from_pretrained(repo_id)
    model = UdopForConditionalGeneration.from_pretrained(repo_id)

    return model, processor