model_api.py

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

def load_model():
    tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
    return tokenizer, model