jonathanjordan21
commited on
Commit
•
95ef1a7
1
Parent(s):
963bc16
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)
|