File size: 511 Bytes
e037c22 8ae1f68 e037c22 8ae1f68 3cc00a5 8ae1f68 3cc00a5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from flask import Flask, request, jsonify
from model import load_model, predict
app = Flask(__name__)
# Load mock model components
model, tokenizer, summarizer = load_model()
@app.route("/score", methods=["POST"])
def score():
try:
data = request.get_json()
result = predict(data, model, tokenizer, summarizer)
return jsonify(result), 200
except Exception as e:
return jsonify({"error": str(e)}), 500
if __name__ == "__main__":
app.run(host="0.0.0.0", port=7860)
|