Om-Shandilya commited on
Commit
8ba278c
·
1 Parent(s): 71e06b8

Revert back to using precomputed TF-IDF vectorizers and Matrices

Browse files
Files changed (1) hide show
  1. backend/main.py +1 -1
backend/main.py CHANGED
@@ -35,7 +35,7 @@ async def lifespan(app: FastAPI):
35
  # Loading TF-IDF Models
36
  print("📂 Loading pre-computed TF-IDF models...")
37
  ml_models["applicant_vectorizer"] = load_tfidf_vectorizer(repo_id="Om-Shandilya/resume-matcher-tfidf", filename="applicant/job_vectorizer.pkl", local_vectorizer_path=None)
38
- ml_models["applicant_matrix"] = load_tfidf_matrix(repo_id="Om-Shandilya/resume-matcher-tfidf", matrix_filename="applicant/job_matrix.npz", local_matrix_path=None)
39
  ml_models["recruiter_vectorizer"] = load_tfidf_vectorizer(repo_id="Om-Shandilya/resume-matcher-tfidf", filename="recruiter/combined_vectorizer.pkl", local_vectorizer_path=None)
40
  ml_models["tfidf_job_df"] = load_job_titles(repo_id='Om-Shandilya/resume-matcher-tfidf', filename='applicant/tfidf_job_titles.csv', local_path=None)
41
 
 
35
  # Loading TF-IDF Models
36
  print("📂 Loading pre-computed TF-IDF models...")
37
  ml_models["applicant_vectorizer"] = load_tfidf_vectorizer(repo_id="Om-Shandilya/resume-matcher-tfidf", filename="applicant/job_vectorizer.pkl", local_vectorizer_path=None)
38
+ ml_models["applicant_matrix"] = load_tfidf_matrix(repo_id="Om-Shandilya/resume-matcher-tfidf", filename="applicant/job_matrix.npz", local_matrix_path=None)
39
  ml_models["recruiter_vectorizer"] = load_tfidf_vectorizer(repo_id="Om-Shandilya/resume-matcher-tfidf", filename="recruiter/combined_vectorizer.pkl", local_vectorizer_path=None)
40
  ml_models["tfidf_job_df"] = load_job_titles(repo_id='Om-Shandilya/resume-matcher-tfidf', filename='applicant/tfidf_job_titles.csv', local_path=None)
41