Update app.py
Browse files
app.py
CHANGED
@@ -110,7 +110,7 @@ def prompt_model(model_name, system_prompt, user_prompt, sql_schema):
|
|
110 |
max_new_tokens=1000)
|
111 |
|
112 |
messages = [
|
113 |
-
{"role": "system", "content": system_prompt.format(schema=
|
114 |
{"role": "user", "content": user_prompt},
|
115 |
{"role": "assistant", "content": ""}
|
116 |
]
|
|
|
110 |
max_new_tokens=1000)
|
111 |
|
112 |
messages = [
|
113 |
+
{"role": "system", "content": system_prompt.format(schema=sql_schema)},
|
114 |
{"role": "user", "content": user_prompt},
|
115 |
{"role": "assistant", "content": ""}
|
116 |
]
|