shaheerawan3 commited on
Commit
446f194
·
verified ·
1 Parent(s): 5daf580

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +6 -4
model.py CHANGED
@@ -1,11 +1,13 @@
1
  from transformers import pipeline
2
 
3
- # Load Hugging Face model
4
  model = pipeline("text-generation", model="gpt2")
5
 
6
  def generate_answer(job_title, required_skills):
7
- prompt = (f"I am applying for the role of {job_title}. "
8
- f"My skills include {required_skills}. "
9
- f"Why should the company hire me?")
 
 
10
  result = model(prompt, max_length=150, num_return_sequences=1)
11
  return result[0]['generated_text']
 
1
  from transformers import pipeline
2
 
3
+ # Load Hugging Face Model
4
  model = pipeline("text-generation", model="gpt2")
5
 
6
  def generate_answer(job_title, required_skills):
7
+ prompt = (
8
+ f"I am applying for the role of {job_title}. "
9
+ f"My skills include {required_skills}. "
10
+ f"Why should the company hire me?"
11
+ )
12
  result = model(prompt, max_length=150, num_return_sequences=1)
13
  return result[0]['generated_text']