from flask import Flask, request, jsonify | |
from model import load_model, predict | |
app = Flask(__name__) | |
# Load mock model components | |
model, tokenizer, summarizer = load_model() | |
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) | |