Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 = "
|
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
|