cmagganas commited on
Commit
1cf93c7
·
1 Parent(s): 49990fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -16,9 +16,9 @@ tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
16
  model = PeftModel.from_pretrained(model, peft_model_id)
17
 
18
 
19
- def make_inference(job_posting):
20
  batch = tokenizer(
21
- f"Write a Cover Letter for this ### Job Posting\n\n```\n{job_posting}\n```",
22
  return_tensors="pt",
23
  )
24
 
@@ -35,9 +35,10 @@ if __name__ == "__main__":
35
  gr.Interface(
36
  make_inference,
37
  [
 
38
  gr.inputs.Textbox(lines=10, label="Job Posting")
39
  ],
40
- gr.outputs.Textbox(label="Job Posting"),
41
  title="CoverLetter-GenAI-adapter",
42
  description="Write a cover letter for you based on job description.",
43
  ).launch()
 
16
  model = PeftModel.from_pretrained(model, peft_model_id)
17
 
18
 
19
+ def make_inference(my_cover_letter, job_posting):
20
  batch = tokenizer(
21
+ f"Adapt my Cover Letter ```\n{my_cover_letter}\n```\nto this Job Posting\n```\n{job_posting}\n```",
22
  return_tensors="pt",
23
  )
24
 
 
35
  gr.Interface(
36
  make_inference,
37
  [
38
+ gr.inputs.Textbox(lines=5, label="My Cover Letter"),
39
  gr.inputs.Textbox(lines=10, label="Job Posting")
40
  ],
41
+ gr.outputs.Textbox(label="Cover Letter"),
42
  title="CoverLetter-GenAI-adapter",
43
  description="Write a cover letter for you based on job description.",
44
  ).launch()