updated

backend/services/resume_parser.py

@@ -7,15 +7,12 @@ from docx import Document
 from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
 
 # --------------------
-# Load Resume NER Model
+# Load PyTorch Resume NER Model
 # --------------------
-MODEL_NAME = "Ioana23/bert-finetuned-resumes-ner"
+MODEL_NAME = "manishiitg/resume-ner"
 
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
-from transformers import TFAutoModelForTokenClassification
-
-model = TFAutoModelForTokenClassification.from_pretrained(MODEL_NAME, from_tf=True)
-
+model = AutoModelForTokenClassification.from_pretrained(MODEL_NAME)
 ner_pipeline = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="simple")
 
 # --------------------