Update app.py

app.py

@@ -1,20 +1,18 @@
 from flask import Flask, request, jsonify
-from model import load_model, predict_deal_score
-import logging
+from model import load_model, predict
 
 app = Flask(__name__)
 
-# Load model once at startup
+# Load mock model components
 model, tokenizer, summarizer = load_model()
 
 @app.route("/score", methods=["POST"])
 def score():
     try:
         data = request.get_json()
-        result = predict_deal_score(data, model, tokenizer, summarizer)
+        result = predict(data, model, tokenizer, summarizer)
         return jsonify(result), 200
     except Exception as e:
-        logging.exception("Prediction error")
         return jsonify({"error": str(e)}), 500
 
 if __name__ == "__main__":