Roxanne-WANG commited on
Commit
c10b742
·
1 Parent(s): 13ee483
Files changed (1) hide show
  1. text2sql.py +1 -0
text2sql.py CHANGED
@@ -16,6 +16,7 @@ from tqdm import tqdm
16
  import torch
17
  from transformers import AutoModelForCausalLM, AutoTokenizer
18
  from whoosh import index
 
19
 
20
  from utils.db_utils import (
21
  get_db_schema,
 
16
  import torch
17
  from transformers import AutoModelForCausalLM, AutoTokenizer
18
  from whoosh import index
19
+ import sqlparse
20
 
21
  from utils.db_utils import (
22
  get_db_schema,