predict.py

import tensorflow as tf
import numpy as np

from preprocessing import preprocess
from labels import classes


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


def predict_image(image):

    image = preprocess(image)

    prediction = model.predict(image)[0]

    results = {}

    for i,prob in enumerate(prediction):

        results[classes[i]] = float(prob)

    return results