Spaces:

keerthanaBasavaraj
/

nlp-to-sql

Sleeping

keerthanaBasavaraj commited on Jan 24

Commit

db38d1c

1 Parent(s): a13400a

Update model to chatdb/natural-sql-7b

Files changed (1) hide show

sql_query_generator/generator.py CHANGED Viewed

@@ -1,6 +1,6 @@
 from transformers import AutoTokenizer, AutoModelForCausalLM
-def load_model(model_name="premai-io/prem-1B-SQL"):
     """
     Loads the SQL generation model and tokenizer from Hugging Face.
     """

 from transformers import AutoTokenizer, AutoModelForCausalLM
+def load_model(model_name="chatdb/natural-sql-7b"):
     """
     Loads the SQL generation model and tokenizer from Hugging Face.
     """