Сохранение изображения в виде файла ошибки (0 байт) в Flask - PullRequest
0 голосов
/ 24 апреля 2020

Когда я сохраняю изображение, используя flask, оно сохраняется как файл с 0 байтами

from flask import Flask, render_template, redirect, request, url_for
from inference import predict_breed_transfer
import os
import io
from PIL import Image
from werkzeug.utils import secure_filename

UPLOAD_FOLDER = './static'
ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg'}

app = Flask(__name__)
app.config['SECRET_KEY'] = 'super secret key'
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER

def allowed_file(filename):
    return '.' in filename and \
           filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS

@app.route('/')
def hello():
    return render_template("index.html")

@app.route('/', methods = ['POST'])
def submit_data():
    if request.method == 'POST':
        file = request.files['file']
        if file and allowed_file(file.filename):
            filename = secure_filename(file.filename)
            image = file.read()
            predicted_breed = predict_breed_transfer(image_bytes = image)
            imag = Image.open(io.BytesIO(image))
            imag.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
            result_dic = {
                'im' : filename,
                'breed' : predicted_breed
            }
            return render_template('index.html', dog=result_dic)
        return render_template("index.html")

if __name__ == '__main__':
    app.run()

Пожалуйста, помогите мне с этим. Кроме того, для вашей помощи, этот проект размещен на https://dog-identification.herokuapp.com/

Снимок экрана свойств изображения, который создается после сохранения изображения

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...