LeonceNsh commited on
Commit
238955b
·
verified ·
1 Parent(s): e57303b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -74,12 +74,12 @@ def parse_query(nl_query):
74
 
75
  try:
76
  response = openai.chat.completions.create(
77
- model="gpt-4",
78
  messages=messages,
79
  temperature=0,
80
  max_tokens=150,
81
  )
82
- sql_query = response.choices[0].message['content'].strip()
83
  return sql_query
84
  except Exception as e:
85
  return f"Error generating SQL query: {e}"
 
74
 
75
  try:
76
  response = openai.chat.completions.create(
77
+ model="gpt-4o-mini",
78
  messages=messages,
79
  temperature=0,
80
  max_tokens=150,
81
  )
82
+ sql_query = response.choices[0].message.content.strip()
83
  return sql_query
84
  except Exception as e:
85
  return f"Error generating SQL query: {e}"