部署 cnn 模型时发生 heroku 500(内部服务器错误)

时间:2021-02-16 03:33:00

标签: python heroku

我已经使用 Flask 和 Heroku 部署了一个 CNN 模型。在代码模型的一部分中,模型需要一个图像,该图像将保存在一个文件夹名称上传中,然后图像的路径被提供给模型进行预测。当我在本地机器上运行模型时,它可以正常工作,但在 heroku 上显示 500 内部错误,可能是因为上传文件问题。这是我上传功能代码的一瞥。

def upload():
    if request.method == 'POST':
        # Get the file from post request
        f = request.files['file']

        # Save the file to ./uploads
        basepath = os.path.dirname(__file__)
        file_path = os.path.join(
            basepath, 'uploads', secure_filename(f.filename))
        f.save(file_path)
        print(file_path)

        # Make prediction
        preds = model_predict(file_path)
        return preds

据我所知,我必须对模型预测的方式或其他方式进行一些更改。如果您知道解决方案,请帮助我。

该图像仅在一个实例中是必需的,因此没有使用它,因此如果有一种方法可以使用我可以虚拟加载图像然后将其删除,那就更好了

Heroku 网址:https://banana-ripeness-detection.herokuapp.com/ Github 网址:https://github.com/Abhayparashar31/brd

当图像上传并发送进行预测时的主要内容和 500 内部错误。

提前致谢

0 个答案:

没有答案