jonathanjordan21 commited on
Commit
f44be29
·
verified ·
1 Parent(s): a5f7e84

Update parser.py

Browse files
Files changed (1) hide show
  1. parser.py +1 -2
parser.py CHANGED
@@ -1,5 +1,4 @@
1
  from transformers import AutoModelForCausalLM, AutoTokenizer
2
- import json
3
 
4
  model_name = "Qwen/Qwen3-0.6B"
5
 
@@ -63,4 +62,4 @@ def parse_resume(resume, parsing):
63
 
64
  thinking_content = tokenizer.decode(output_ids[:index], skip_special_tokens=True).strip("\n")
65
  content = tokenizer.decode(output_ids[index:], skip_special_tokens=True).strip("\n")
66
- return thinking_content, json.loads(content)
 
1
  from transformers import AutoModelForCausalLM, AutoTokenizer
 
2
 
3
  model_name = "Qwen/Qwen3-0.6B"
4
 
 
62
 
63
  thinking_content = tokenizer.decode(output_ids[:index], skip_special_tokens=True).strip("\n")
64
  content = tokenizer.decode(output_ids[index:], skip_special_tokens=True).strip("\n")
65
+ return thinking_content, content