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 |
|