ats / app.py
missravneetkaur's picture
Update app.py
28e12b6 verified
raw
history blame contribute delete
632 Bytes
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")
@app.route("/", methods=["GET", "POST"])
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)