ArthurLin commited on
Commit
a2cd8f6
·
verified ·
1 Parent(s): 742b7ae

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -15,7 +15,7 @@ def run_single_similarity(jd_path, cv_path, SECTION_WEIGHTS, model_id="sentence-
15
  weighted_scores = sum(scores[k] * SECTION_WEIGHTS.get(k, 0) for k in scores)
16
  return round(float(weighted_scores), 4), scores
17
 
18
- def run_batch_similarity(jd_path, cv_input, SECTION_WEIGHTS, model_id="sentence-transformers/all-MiniLM-L6-v2", llm_model="meta-ll ama/Meta-Llama-3-8B-Instruct", pipe=None):
19
 
20
  jd_text = load_text(jd_path)
21
  jd_embeddings, _ = generate_summary_and_embedding(jd_text, text_type="job_description", model_id=model_id, llm_model=llm_model, pipe=pipe)
 
15
  weighted_scores = sum(scores[k] * SECTION_WEIGHTS.get(k, 0) for k in scores)
16
  return round(float(weighted_scores), 4), scores
17
 
18
+ def run_batch_similarity(jd_path, cv_input, SECTION_WEIGHTS, model_id="sentence-transformers/all-MiniLM-L6-v2", llm_model="meta-llama/Meta-Llama-3-8B-Instruct", pipe=None):
19
 
20
  jd_text = load_text(jd_path)
21
  jd_embeddings, _ = generate_summary_and_embedding(jd_text, text_type="job_description", model_id=model_id, llm_model=llm_model, pipe=pipe)