我有一个大文件,我们称之为 machine_learning_model.hdf5。每次点击发布请求端点时,我都会将其加载到我的应用程序中。伪代码如下所示:
def post(self):
model = load_model('./machine_learning_model.hdf5')
return( model.predict())
然而,事实上,每次端点被命中时我都会加载文件,这会导致问题。一般来说,在初始化时将大文件加载到 Flask 应用程序中的正确方法是什么,以便各个端点可以使用所述文件中的代码。
您可以在应用程序启动时加载它并绑定到 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()
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)