如何在Flask应用程序中的初始化时加载文件

时间:2018-11-05 21:23:44

标签: flask flask-restplus

我有一个大文件,我们称它为machine_learning_model.hdf5。每次点击后请求端点时,我都会将其加载到我的应用程序中。伪代码如下所示:

def post(self):
  model = load_model('./machine_learning_model.hdf5')
  return( model.predict())

但是,每次碰到端点都会导致我加载文件这一事实。通常,在初始化时将大文件加载到flask应用程序中的正确方法是什么,以便各个端点可以使用该文件中的代码。

1 个答案:

答案 0 :(得分:1)

您可以在应用程序启动时加载它并绑定到flask应用程序对象:

# app.py
app = Flask(__name__)
app.some_model = load_model('./machine_learning_model.hdf5')

# handlers.py
from flask import current_app

def post(self):
    return( current_app.some_model.predict())