Views
No views yet
1from transformers import GenerationConfig
2from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3from transformers import GenerationConfig
4import nltk
5nltk.download('punkt')
6max_source_length=512
7tokenizer = AutoTokenizer.from_pretrained("Hariharavarshan/Cover_genie")
8model = AutoModelForSeq2SeqLM.from_pretrained("Hariharavarshan/Cover_genie")
9JD='''<Job description Text>'''
10resume_text= '''<Resume Text>'''
11final_text="give me a cover letter based on the a job description and a resume. Job description:"+JD +" Resume:"+ resume_text
12generation_config = GenerationConfig.from_pretrained("google/flan-t5-large",temperature=2.0)
13inputs = tokenizer(final_text, max_length=max_source_length, truncation=True, return_tensors="pt")
14output = model.generate(**inputs, num_beams=3, do_sample=True, min_length=1000,
15 max_length=10000,generation_config=generation_config,num_return_sequences=3)
16decoded_output = tokenizer.batch_decode(output, skip_special_tokens=True)[0]
17generated_Coverletter = nltk.sent_tokenize(decoded_output.strip())