Update README.md

README.md

@@ -54,7 +54,7 @@ Here is how to load and use the model:
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_name = "llama-2-7b-sft-lora-4bit-float16"
+model_name = "Vanessasml/cyber-risk-llama-2-7b"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
@@ -66,7 +66,7 @@ The insurer polled over 1000 adults over the age of 45 in the latest update to i
 Further, 6% said they had actually fallen victim to such an online attack, amounting to around 1.2 million adults. \
 Some 22% more people it surveyed had been targeted by ...	
 """
-pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=2000)
+pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=200)
 # To generate text:
 result = pipe(f"<s>[INST] {prompt} [/INST]")
 print(result[0]['generated_text'])