Update app.py
Browse files
app.py
CHANGED
|
@@ -35,10 +35,8 @@ def generate(content):
|
|
| 35 |
)
|
| 36 |
|
| 37 |
messages = [
|
| 38 |
-
{
|
| 39 |
-
|
| 40 |
-
"role": "user", "content": content
|
| 41 |
-
},
|
| 42 |
]
|
| 43 |
inputs = tokenizer.apply_chat_template(
|
| 44 |
messages,
|
|
|
|
| 35 |
)
|
| 36 |
|
| 37 |
messages = [
|
| 38 |
+
{ "role": "system", "content": SYSTEM_PROMPT },
|
| 39 |
+
{ "role": "user", "content": content },
|
|
|
|
|
|
|
| 40 |
]
|
| 41 |
inputs = tokenizer.apply_chat_template(
|
| 42 |
messages,
|