Views
No views yet
pip install mlx-lm1from mlx_lm import load, generate
2
3model, tokenizer = load("juooo1117/skill-extractor-llama-3.1-8B")
4
5instruction = "Extract job-related professional competencies in English JSON format from the given Korean self-introduction."
6resume_text = "여기에 자기소개서 내용을 넣으세요."
7
8messages = [{"role": "user", "content": f"{instruction}\n\n{resume_text}"}]
9prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
10prompt += '{"competencies": ["'
11
12response = generate(model=model, tokenizer=tokenizer, prompt=prompt, max_tokens=500, verbose=True)
13final_output = '{["' + response