Spaces:
Runtime error
Runtime error
# 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) | |