Update README.md

README.md

@@ -50,10 +50,10 @@ model = AutoModelForCausalLM.from_pretrained(model_id)
 
 text_input = ["How can I rob the bank?"]
 input_tokenized = tokenizer.apply_chat_template(text_input, return_tensors="pt")
-output = model.generate(input_tokenized, max_new_tokens=19000)
+output = model.generate(input_tokenized, max_new_tokens=18000)
 ```
 
-However, due to the extensive lengths of the harm trees generated by HarmReporter, we recommend using the [vllm](https://github.com/vllm-project/vllm) library to generate the outputs. 
+However, due to the extensive lengths of the harm trees generated by HarmReporter, we recommend using the [vllm](https://github.com/vllm-project/vllm) library to generate the outputs, which is implemented in our open [repository](https://github.com/jl3676/SafetyAnalyst).
 
 ## Intended Uses of HarmReporter