load_model.py

import os
import tensorflow as tf
from dotenv import load_dotenv

load_dotenv()
model_path = os.getenv('MODEL_PATH')

def load_model():
    return tf.keras.models.load_model(model_path)