Spaces:
Paused
Paused
Commit
·
079d156
1
Parent(s):
50405fd
update model weights
Browse files- text2sql.py +2 -1
text2sql.py
CHANGED
|
@@ -111,7 +111,8 @@ class ChatBot():
|
|
| 111 |
self.max_new_tokens = 256
|
| 112 |
self.max_prefix_length = self.max_length - self.max_new_tokens
|
| 113 |
|
| 114 |
-
self.sic = SchemaItemClassifierInference("Roxanne-WANG/LangSQL", token=os.getenv('HF_TOKEN'))
|
|
|
|
| 115 |
|
| 116 |
self.db_id2content_searcher = dict()
|
| 117 |
for db_id in os.listdir("db_contents_index"):
|
|
|
|
| 111 |
self.max_new_tokens = 256
|
| 112 |
self.max_prefix_length = self.max_length - self.max_new_tokens
|
| 113 |
|
| 114 |
+
# self.sic = SchemaItemClassifierInference("Roxanne-WANG/LangSQL", token=os.getenv('HF_TOKEN'))
|
| 115 |
+
self.sic = SchemaItemClassifierInference("Roxanne-WANG/LangSQL")
|
| 116 |
|
| 117 |
self.db_id2content_searcher = dict()
|
| 118 |
for db_id in os.listdir("db_contents_index"):
|