1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = 'Jwalit/kyc-document-corner-detector-v2'
4tokenizer = AutoTokenizer.from_pretrained(model_id)
5model = AutoModelForCausalLM.from_pretrained(model_id)AutoModelForCausalLM with the appropriate AutoModel class.