Spaces:
Sleeping
Sleeping
| from flask import Flask, render_template, request | |
| import joblib | |
| import os | |
| app = Flask(__name__) | |
| # Load models | |
| tfidf_vectorizer = joblib.load("model/tfidf_vectorizer.pkl") | |
| classifier = joblib.load("model/resume_classifier.pkl") | |
| def home(): | |
| prediction = None | |
| if request.method == "POST": | |
| resume_text = request.form["resume"] | |
| vectorized = tfidf_vectorizer.transform([resume_text]) | |
| prediction = classifier.predict(vectorized)[0] | |
| return render_template("index.html", prediction=prediction) | |
| if __name__ == "__main__": | |
| app.run(host="0.0.0.0", port=7860) | |