fix: typo in python code of summarization example

#10
Files changed (1) hide show
  1. README.md +1 -1
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": system_prompt_rewrite}, {"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)
 
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)