From fd069e9c8cafd13db6df2f08e46f9c570991d6bb Mon Sep 17 00:00:00 2001 From: r0r-5chach Date: Sat, 8 Jun 2024 20:19:05 +0100 Subject: [PATCH] python prediction code --- src/model/model.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) 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)