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
|