ATS_score.py

from sentence_transformers import SentenceTransformer, util

def calculateATSscore(resume_data, job_description):
    model = SentenceTransformer('paraphrase-MiniLM-L6-v2')  # Pre-trained BERT model
    resume_embedding = model.encode(resume_data)
    job_desc_embedding = model.encode(job_description)
    similarity_value = util.cos_sim(resume_embedding, job_desc_embedding)
    return similarity_value.item()