vidulpanickan commited on
Commit
bc6e865
·
verified ·
1 Parent(s): 910ce57

Update backend model

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -6,7 +6,7 @@ from huggingface_hub import InferenceClient
6
 
7
  HF_TOKEN = os.environ.get("HF_TOKEN")
8
  client = InferenceClient(
9
- model="Qwen/Qwen2.5-Coder-32B-Instruct",
10
  token=HF_TOKEN,
11
  )
12
 
@@ -34,7 +34,8 @@ SQL RULES:
34
  - Age filters: use a numeric comparison on the age column directly (e.g. age > 50).
35
  - Counts: use COUNT(*) or COUNT(column). Alias it clearly, e.g. AS num_patients.
36
  - INSERT, UPDATE, DELETE, CREATE, DROP, ALTER are all allowed — the user owns their database.
37
-
 
38
  FEW-SHOT EXAMPLES:
39
 
40
  Schema:
 
6
 
7
  HF_TOKEN = os.environ.get("HF_TOKEN")
8
  client = InferenceClient(
9
+ model="Snowflake/arctic-text2sql-r1-32b",
10
  token=HF_TOKEN,
11
  )
12
 
 
34
  - Age filters: use a numeric comparison on the age column directly (e.g. age > 50).
35
  - Counts: use COUNT(*) or COUNT(column). Alias it clearly, e.g. AS num_patients.
36
  - INSERT, UPDATE, DELETE, CREATE, DROP, ALTER are all allowed — the user owns their database.
37
+ - Identifier quoting: always wrap table and column names in double quotes if they start with a digit or contain special characters (e.g. "2b_concept", "my-column").
38
+ ...
39
  FEW-SHOT EXAMPLES:
40
 
41
  Schema: