from flask import Flask, request, jsonify
import tensorflow as tf
import numpy as np

model = tf.keras.models.load_model("model2.keras")

app = Flask(__name__)

@app.route("/predict", methods=["POST"])
def predict():
    data = request.json
    inputs = np.array(data["features"]).reshape(1, -1)
    prediction = model.predict(inputs)
    return jsonify({"grade": int(prediction.argmax())})

app.run(host="0.0.0.0", port=7860)