MuntasirHossain commited on
Commit
eed94cf
1 Parent(s): a193dc2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -39,7 +39,7 @@ def generate(input_text):
39
  ]
40
  prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
41
  outputs = llm(prompt)
42
- return print(outputs[0]["generated_text"][len(prompt):])
43
 
44
  generate("What is a large language model?")
45
  ````
 
39
  ]
40
  prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
41
  outputs = llm(prompt)
42
+ return outputs[0]["generated_text"][len(prompt):]
43
 
44
  generate("What is a large language model?")
45
  ````