bstraehle commited on
Commit
cda682a
1 Parent(s): 5b45741

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=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
  ]