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

model = tf.keras.models.load_model("model.keras")

classes = ["cats", "dogs", "women", "men"]

img = Image.open("tests/men_1.jpg")
img = img.convert("RGB")
img = img.resize((224, 224))

arr = np.array(img) / 255.0
arr = np.expand_dims(arr, axis=0)

pred = model.predict(arr, verbose=0)

idx = np.argmax(pred)

print("Class:", classes[idx])
print("Probabilities:", pred[0])