Textwizai / load_model.py
Erfan11's picture
Update load_model.py
8bad412 verified
raw
history blame
186 Bytes
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)