karthikmn's picture
Update app.py
8ae1f68 verified
raw
history blame
511 Bytes
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)