jonathanjordan21 commited on
Commit
7c6155e
·
verified ·
1 Parent(s): 0ccbb95

Update ranker.py

Browse files
Files changed (1) hide show
  1. ranker.py +17 -0
ranker.py CHANGED
@@ -55,3 +55,20 @@ def rank_resume(job_description, resumes):
55
  scores = compute_logits(inputs)
56
 
57
  return resumes,scores
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
55
  scores = compute_logits(inputs)
56
 
57
  return resumes,scores
58
+
59
+
60
+
61
+ def rank_resume_multi(job_description, resumes):
62
+ task = "Given a Job description, retrieve relevant resume that is suitable for the job"
63
+
64
+ queries = [resumes] * len(job_description)
65
+ documents = job_description
66
+ # documents = [job_description] * len(queries)
67
+
68
+
69
+ pairs = [format_instruction(task, query, doc) for query, doc in zip(queries, documents)]
70
+
71
+ inputs = process_inputs(pairs)
72
+ scores = compute_logits(inputs)
73
+
74
+ return resumes,scores