Commit
•
eed94cf
1
Parent(s):
a193dc2
Update README.md
Browse files
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
|
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 |
````
|