bank_transaction / python.txt
leynessa's picture
Upload 8 files
ff52cdd verified
raw
history blame contribute delete
624 Bytes
# run_api.py
from flask import Flask, request, jsonify
import joblib
app = Flask(__name__)
model = joblib.load("logreg_model.pkl")
vectorizer = joblib.load("tfidf_vectorizer.pkl")
@app.route("/classify", methods=["POST"])
def classify():
data = request.get_json()
text = data.get("purpose_text", "")
if not text:
return jsonify({"error": "Empty input"}), 400
cleaned = preprocess_text(text)
vec = vectorizer.transform([cleaned])
prediction = model.predict(vec)[0]
return jsonify({"predicted_type": prediction})
if __name__ == "__main__":
app.run(debug=True)