Spaces:
Running
Running
File size: 415 Bytes
fc45a55 bd3211b fc45a55 7e4deac |
1 2 3 4 5 6 7 8 9 10 11 12 |
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()
|