我想要一个sanic部署机器学习模型的代码
时间: 2023-02-05 21:47:02 浏览: 117
好的,首先需要安装sanic和其他所需的Python包,可以使用以下命令进行安装:
```
pip install sanic tensorflow numpy scipy
```
然后,可以使用以下代码示例来构建基于sanic的机器学习模型API:
```
import tensorflow as tf
import numpy as np
import sanic
from sanic import response
# 加载机器学习模型
model = tf.keras.models.load_model('model.h5')
# 创建sanic应用程序
app = sanic.Sanic()
@app.route("/predict", methods=["POST"])
async def predict(request):
# 获取输入数据
data = request.json
# 使用模型进行预测
prediction = model.predict(data)
# 返回预测结果
return response.json({"prediction": prediction.tolist()})
if __name__ == "__main__":
app.run(host="0.0.0.0", port=8000)
```
这段代码中,我们使用了sanic的路由装饰器`@app.route`来定义路由,然后使用`request.json`获取输入数据,并使用模型进行预测。最后,使用`response.json`函数将预测结果作为JSON响应返回给客户端。
要启动应用程序,只需在命令行中运行`python app.py`,然后在浏览器中访问`http://localhost:8000/predict`即可。
希望这个示例对你有帮助。
阅读全文