TKM03 commited on
Commit
99e48e5
·
verified ·
1 Parent(s): 8bfcb85

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -8,7 +8,8 @@ from collections import Counter
8
  ner_pipeline = pipeline("ner", model="dslim/bert-base-NER", tokenizer="dslim/bert-base-NER", aggregation_strategy="simple")
9
 
10
  # Load text classification model (replace with a job-role classifier if available)
11
- text_classifier = pipeline("text-classification", model="MoritzLaurer/bert-multilingual-passage-reranking-msmarco")
 
12
 
13
  def clean_resume_text(text):
14
  text = re.sub(r'http\S+', ' ', text)
 
8
  ner_pipeline = pipeline("ner", model="dslim/bert-base-NER", tokenizer="dslim/bert-base-NER", aggregation_strategy="simple")
9
 
10
  # Load text classification model (replace with a job-role classifier if available)
11
+ text_classifier = pipeline("text-classification", model="khaimait/job-title-classification-bert")
12
+
13
 
14
  def clean_resume_text(text):
15
  text = re.sub(r'http\S+', ' ', text)