Update README.md

README.md

@@ -87,11 +87,12 @@ model = PeftModel.from_pretrained(model, "TuningAI/Llama2_7B_Cover_letter_genera
 tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-hf" , trust_remote_code=True)
 tokenizer.pad_token = tokenizer.eos_token
 tokenizer.padding_side = "right"
+Instruction = "Given a user's information about the target job, you will generate a Cover letter for this job based on this information."
 while 1:
   input_text = input(">>>")
   logging.set_verbosity(logging.CRITICAL)
-  prompt = f"### Instruction\n{system_message}.\n ###Input \n\n{input_text}. ### Output:"
-  pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer,max_length=512)
+  prompt = f"### Instruction\n{Instruction}.\n ###Input \n\n{input_text}. ### Output:"
+  pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer,max_length=400)
   result = pipe(prompt)
   print(result[0]['generated_text'].replace(prompt, ''))
 ```