Spaces:
Sleeping
Sleeping
Update inference.py
Browse files- 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-
|
| 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)
|