husseinelsaadi commited on
Commit
7ec2f5a
·
1 Parent(s): 7f0bef9
Files changed (1) hide show
  1. backend/services/resume_parser.py +4 -1
backend/services/resume_parser.py CHANGED
@@ -12,7 +12,10 @@ from transformers import AutoTokenizer, AutoModelForTokenClassification, pipelin
12
  MODEL_NAME = "Ioana23/bert-finetuned-resumes-ner"
13
 
14
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
15
- model = AutoModelForTokenClassification.from_pretrained(MODEL_NAME)
 
 
 
16
  ner_pipeline = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="simple")
17
 
18
  # --------------------
 
12
  MODEL_NAME = "Ioana23/bert-finetuned-resumes-ner"
13
 
14
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
15
+ from transformers import TFAutoModelForTokenClassification
16
+
17
+ model = TFAutoModelForTokenClassification.from_pretrained(MODEL_NAME, from_tf=True)
18
+
19
  ner_pipeline = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="simple")
20
 
21
  # --------------------