Spaces:
Build error
Build error
File size: 435 Bytes
5ac006c 446f194 5ac006c 446f194 5ac006c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | from transformers import pipeline
# Load Hugging Face Model
model = pipeline("text-generation", model="gpt2")
def generate_answer(job_title, required_skills):
prompt = (
f"I am applying for the role of {job_title}. "
f"My skills include {required_skills}. "
f"Why should the company hire me?"
)
result = model(prompt, max_length=150, num_return_sequences=1)
return result[0]['generated_text']
|