import numpy as np
from PIL import Image
from tensorflow import keras

saved_model_path = 'resources/model.h5'
model = keras.models.load_model(saved_model_path)


image_path = 'resources/indeks-fragile.png'
image_size = (32, 32)

image = Image.open(image_path).convert('L')
image = image.resize(image_size)
image_array = np.array(image) / 255.0
input_data = np.expand_dims(image_array, axis=0)


predictions = model.predict(input_data)
predicted_class = np.argmax(predictions, axis=1)


class_names = ['fragile', 'priority', 'skull']
predicted_label = class_names[predicted_class[0]]

print('Predicted class:', predicted_label)