Anusha-TS commited on
Commit
e3d1e5e
·
verified ·
1 Parent(s): f30c79f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ def load_model():
59
  """
60
  Loads the SQL generation model and tokenizer from Hugging Face.
61
  """
62
- model_name = "premai-io/prem-1B-SQL"
63
  tokenizer = AutoTokenizer.from_pretrained(model_name)
64
  model = AutoModelForCausalLM.from_pretrained(model_name)
65
  return tokenizer, model
 
59
  """
60
  Loads the SQL generation model and tokenizer from Hugging Face.
61
  """
62
+ model_name = "defog/sqlcoder-7b"
63
  tokenizer = AutoTokenizer.from_pretrained(model_name)
64
  model = AutoModelForCausalLM.from_pretrained(model_name)
65
  return tokenizer, model