File size: 828 Bytes
47a377e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
import torch
print(">> Loading TinyLlama Model...")
model_path = "./tinyllama-resume-bot"
tokenizer = AutoTokenizer.from_pretrained(model_path)
model = AutoModelForCausalLM.from_pretrained(model_path)
generator = pipeline(
"text-generation",
model=model,
tokenizer=tokenizer,
device=0 if torch.cuda.is_available() else -1
)
def tailor_resume(job_description):
prompt = f"""
Given the job description below, generate a tailored resume snippet that matches the role:
Job Description:
{job_description}
Tailored Resume:
"""
result = generator(prompt, max_new_tokens=300, do_sample=True, temperature=0.7)[0]['generated_text']
return result.split("Tailored Resume:")[-1].strip()
|