#Wyświetlanie sklasyfikowanej etykiety
from pathlib import Path
from PIL import Image, ImageOps
import matplotlib.pyplot as plt
import numpy as np

# Testowanie grafik z folderu
for path in Path("testowe").rglob("*.png"):
    image = Image.open(path).convert('L')
    image = ImageOps.invert(image)
    image = image.resize((28, 28))
    image = np.array(image).reshape(1, 28, 28, 1) / 255.0

    plt.imshow(image.reshape(28, 28), cmap='gray')
    plt.title('Przetworzony obraz')
    plt.axis('off')
    plt.show()

    prediction = model.predict(image)[0]
    predicted_digit = np.argmax(prediction) 

    print(f"Rozpoznana cyfra: {predicted_digit}")