Нейронка - сервер

[dim222]

Сори за нубный вопрос!
Обучил нейронку, т.к. имеется необходимость в распознавании картинок.
С данным языком я на ВЫ прошу камнями не кидаться

Данным кодом , я поднимаю сервер, и возвращаю текстом предполагаемое на картинке

from __future__ import absolute_import, division, print_function, unicode_literals

from flask import Flask, request

import tensorflow as tf
from tensorflow.keras.preprocessing.image import ImageDataGenerator

import os
import numpy as np
import glob
import shutil
import matplotlib.pyplot as plt

 # директория скрипта
directory = os.path.dirname(os.path.abspath(__file__))
classes = ['airplane', 'bicycle', 'boat', 'car', 'motorbus', 'motorcycle', 'train', 'truck', 'umbrella']

app = Flask(__name__)
@app.route("/file", methods=['GET','POST'])

def file():
    data = request.files['file'].read()
    with open("Image/img.png", mode="wb") as new:
        new.write(data)
 
    validation_dir = directory
    
    image_gen_val = ImageDataGenerator(rescale=1./255)
    
    val_data_gen = image_gen_val.flow_from_directory(batch_size=32,
        directory=validation_dir,
        target_size=(140, 140),
        class_mode='binary')

    model = tf.keras.models.load_model(directory + '/hCaptcha.h5')

    model.summary()

    predictions = model.predict(val_data_gen)
    
    return classes[np.argmax(predictions[0])]

    
if __name__ == '__main__':
    app.run(host='192.168.1.7', port=82)

Но при работе данного кода происходит просто ЖОР оперативы, буквально за пару часов может перевалить за 10 Гб.
Уверен, что проблема в загрузки модуля при каждом запросе.Но по другому у меня не фурычит!
Меня устроит любой рабочий вариант!

Что мне нужно :

- отправил картинку в формате blob

- получил ответ от нейронки

Может кто-нить помочь?