| 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] | |