Update README.md

README.md

@@ -16,7 +16,7 @@ tokenizer.pad_token = tokenizer.eos_token
 tokenizer.padding_side = "right"
 
 prompt = "What is a large language model?"
-pipe = pipeline(task="text-generation", model_name=model, tokenizer=tokenizer, max_length=200)
+pipe = pipeline(task="text-generation", model=model_name, tokenizer=tokenizer, max_length=200)
 result = pipe(f"<s>[INST] {prompt} [/INST]")
 print(result[0]['generated_text'])
 ```