Докер не поддерживается между экземплярами NoneType и int - PullRequest
0 голосов
/ 28 сентября 2019

Я пытаюсь запустить докер-контейнер с моделью Flask. Я уже заметил проблему.Я пытаюсь загрузить предварительно подготовленную модель, используя файл рассола.Однако файл Docker не загружает протравленную модель -> NonType

Если я запускаю приложение фляги без докера, я не получаю никаких ошибок.Файл pickle находится в той же папке, что и файл python, поэтому путь должен быть правильным

Вот мой код

import pickle
from flask import Flask, request
from flasgger import Swagger
import numpy as np
import pandas as pd

with open('./rf.pkl', 'rb') as model_file:
    model = pickle.load(model_file)

app = Flask(__name__)
swagger = Swagger(app)


@app.route('/')
def hello():
    return ('hello docker')

@app.route('/predict')
def predict_iris():

    s_length = request.args.get("s_length")
    s_width = request.args.get("s_width")
    p_length = request.args.get("p_length")
    p_width = request.args.get("p_width")

    prediction = clf.predict(np.array([[s_length, s_width, p_length, p_width]]))
    return str(prediction)

@app.route('/predict_file', methods=["POST"])
def predict_iris_file():

    input_data = pd.read_csv(request.files.get("input_file"), header=None)
    prediction = clf.predict(input_data)
    return str(list(prediction))
if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5000)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...