sidharth-pm commited on
Commit
581fba6
·
verified ·
1 Parent(s): b260a50

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,7 +13,7 @@ sql_pipeline = None
13
 
14
  def load_model():
15
  global tokenizer, sql_pipeline
16
- print("Loading SQLCoder-7B-2 ...")
17
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
18
  model = AutoModelForCausalLM.from_pretrained(
19
  MODEL_ID,
@@ -171,7 +171,7 @@ def create_app():
171
  gr.HTML("""
172
  <div class="title-block">
173
  <h1>SQLCoder Studio</h1>
174
- <p>Natural language to SQL to Results &nbsp;|&nbsp; Powered by defog/sqlcoder-7b-2</p>
175
  <div style="margin-top:0.8rem">
176
  <span class="badge">employees</span>
177
  <span class="badge">departments</span>
 
13
 
14
  def load_model():
15
  global tokenizer, sql_pipeline
16
+ print("microsoft/tapex-large-sql-execution ...")
17
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
18
  model = AutoModelForCausalLM.from_pretrained(
19
  MODEL_ID,
 
171
  gr.HTML("""
172
  <div class="title-block">
173
  <h1>SQLCoder Studio</h1>
174
+ <p>Natural language to SQL to Results &nbsp;|&nbsp; Powered by microsoft/tapex-large-sql-execution</p>
175
  <div style="margin-top:0.8rem">
176
  <span class="badge">employees</span>
177
  <span class="badge">departments</span>