husseinelsaadi commited on
Commit
b336194
·
1 Parent(s): d4b2339
Files changed (1) hide show
  1. backend/services/resume_parser.py +2 -1
backend/services/resume_parser.py CHANGED
@@ -11,7 +11,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
11
  # ===============================
12
  # Load DeepSeek Janus-Pro-7B Model
13
  # ===============================
14
- MODEL_ID = "deepseek-ai/Janus-Pro-7B"
 
15
 
16
  print(f"Loading {MODEL_ID}... (This may take some time on first run)")
17
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
 
11
  # ===============================
12
  # Load DeepSeek Janus-Pro-7B Model
13
  # ===============================
14
+ MODEL_ID = "mistralai/Mistral-7B-Instruct-v0.2"
15
+
16
 
17
  print(f"Loading {MODEL_ID}... (This may take some time on first run)")
18
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)