shaheerawan3's picture
Update model.py
446f194 verified
raw
history blame contribute delete
435 Bytes
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']