diff --git a/src/model/model.py b/src/model/model.py index e69de29..ae38aa9 100644 --- a/src/model/model.py +++ b/src/model/model.py @@ -0,0 +1,12 @@ +import io +import tensorflow as tf +from PIL import Image + + +def predict(model_bytes, image_bytes): + model_file = io.BytesIO(model_bytes) + model = tf.keras.models.load(model_file) + + image = Image.open(io.BytesIO(image_bytes)) + + return model.predict(image)