File size: 328 Bytes
5f10e37 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
import numpy as np
import tensorflow as tf
from app.model.preprocess import preprocess_input_data
model = tf.keras.models.load_model("app/model/model.h5")
def predict_quality(raw_data):
processed = preprocess_input_data(raw_data)
preds = model.predict(processed)
return [int(p > 0.5) for p in preds]
|