fix: typo in python code of summarization example
Browse files
README.md
CHANGED
|
@@ -114,7 +114,7 @@ Hey there! I noticed that the CI isn't passing after your latest commit. Could y
|
|
| 114 |
|
| 115 |
```python
|
| 116 |
system_prompt_summarize = "Provide a concise, objective summary of the input text in up to three sentences, focusing on key actions and intentions without using second or third person pronouns."
|
| 117 |
-
messages = [{"role": "system", "content":
|
| 118 |
input_text=tokenizer.apply_chat_template(messages, tokenize=False)
|
| 119 |
inputs = tokenizer.encode(input_text, return_tensors="pt").to(device)
|
| 120 |
outputs = model.generate(inputs, max_new_tokens=50, temperature=0.2, top_p=0.9, do_sample=True)
|
|
|
|
| 114 |
|
| 115 |
```python
|
| 116 |
system_prompt_summarize = "Provide a concise, objective summary of the input text in up to three sentences, focusing on key actions and intentions without using second or third person pronouns."
|
| 117 |
+
messages = [{"role": "system", "content": system_prompt_summarize}, {"role": "user", "content": INSERT_LONG_EMAIL}]
|
| 118 |
input_text=tokenizer.apply_chat_template(messages, tokenize=False)
|
| 119 |
inputs = tokenizer.encode(input_text, return_tensors="pt").to(device)
|
| 120 |
outputs = model.generate(inputs, max_new_tokens=50, temperature=0.2, top_p=0.9, do_sample=True)
|