import numpy as np
from PIL import Image
from tensorflow.keras.models import load_model
from tensorflow.keras.layers import DepthwiseConv2D


def main():
    print("Загрузка СВОЕЙ МОДЕЛИ ...")
    model = load_model(
        "keras_model.h5",
        compile=False,
        custom_objects={"DepthwiseConv2D": DepthwiseConv2D},
    )
    print("РАБОТА с IMG")
    img = Image.open("test.jpg").resize((224, 224))
    print("TYT")
    img_array = np.array(img) / 255.0
    print("TYT 2")
    img_array = np.expand_dims(img_array, axis=0)

    # predictions = model.predict(img_array)
    # print(predictions)


if __name__ == "__main__":
    main()