teaevo commited on
Commit
cb62d0f
·
1 Parent(s): e030ac0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ chatbot_tokenizer = AutoTokenizer.from_pretrained(chatbot_model_name)
7
  chatbot_model = AutoModelForCausalLM.from_pretrained(chatbot_model_name)
8
 
9
  # Load the SQL model
10
- sql_model_name = "your_sql_model_name" # Replace with the name of the SQL model you want to use
11
  sql_tokenizer = AutoTokenizer.from_pretrained(sql_model_name)
12
  sql_model = AutoModelForCausalLM.from_pretrained(sql_model_name)
13
 
 
7
  chatbot_model = AutoModelForCausalLM.from_pretrained(chatbot_model_name)
8
 
9
  # Load the SQL model
10
+ sql_model_name = "microsoft/tapex-large-sql-execution" # Replace with the name of the SQL model you want to use
11
  sql_tokenizer = AutoTokenizer.from_pretrained(sql_model_name)
12
  sql_model = AutoModelForCausalLM.from_pretrained(sql_model_name)
13