Update main.py
Browse files
main.py
CHANGED
|
@@ -61,7 +61,7 @@ def create_email_prompt(subject, sender, recipients, body):
|
|
| 61 |
"""
|
| 62 |
}
|
| 63 |
]
|
| 64 |
-
prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
| 65 |
return prompt
|
| 66 |
|
| 67 |
|
|
|
|
| 61 |
"""
|
| 62 |
}
|
| 63 |
]
|
| 64 |
+
prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
| 65 |
return prompt
|
| 66 |
|
| 67 |
|