LalitChaudhari3 commited on
Commit
94f744f
·
verified ·
1 Parent(s): 6820f84

Update src/sql_generator.py

Browse files
Files changed (1) hide show
  1. src/sql_generator.py +2 -1
src/sql_generator.py CHANGED
@@ -21,7 +21,8 @@ class SQLGenerator:
21
  print(" ❌ FATAL: API Key missing.")
22
 
23
  # 3. USE THE BEST FREE MODEL
24
- self.model_id = "Qwen/Qwen2.5-Coder-32B-Instruct"
 
25
 
26
  def generate_followup_questions(self, question, sql_query):
27
  return ["Visualize this result", "Export as CSV", "Compare with last year"]
 
21
  print(" ❌ FATAL: API Key missing.")
22
 
23
  # 3. USE THE BEST FREE MODEL
24
+ # Change this line in src/sql_generator.py if 32B gets slow:
25
+ self.model_id = "Qwen/Qwen2.5-Coder-7B-Instruct"
26
 
27
  def generate_followup_questions(self, question, sql_query):
28
  return ["Visualize this result", "Export as CSV", "Compare with last year"]