Commit ·
7b763f7
1
Parent(s): 96d1c47
Update README.md
Browse files
README.md
CHANGED
|
@@ -45,10 +45,6 @@ This model is governed by a custom commercial license from Code Llama. For detai
|
|
| 45 |
## Example Code
|
| 46 |
|
| 47 |
You can use the Code-Llama-2-7B-instruct-text2sql-GGUF model to generate SQL queries from natural language questions, as demonstrated in the following code snippet:
|
| 48 |
-
```cmd
|
| 49 |
-
pip install -q ctransformers
|
| 50 |
-
```
|
| 51 |
-
|
| 52 |
```python
|
| 53 |
import torch
|
| 54 |
from ctransformers import AutoModelForCausalLM
|
|
@@ -69,6 +65,7 @@ question = 'Find the salesperson who made the most sales.'
|
|
| 69 |
prompt = f"[INST] Write SQLite query to answer the following question given the database schema. Please wrap your code answer using ```: Schema: {table} Question: {question} [/INST] Here is the SQLite query to answer to the question: {question}: ``` "
|
| 70 |
|
| 71 |
output = model(prompt)
|
|
|
|
| 72 |
print(output)
|
| 73 |
```
|
| 74 |
|
|
|
|
| 45 |
## Example Code
|
| 46 |
|
| 47 |
You can use the Code-Llama-2-7B-instruct-text2sql-GGUF model to generate SQL queries from natural language questions, as demonstrated in the following code snippet:
|
|
|
|
|
|
|
|
|
|
|
|
|
| 48 |
```python
|
| 49 |
import torch
|
| 50 |
from ctransformers import AutoModelForCausalLM
|
|
|
|
| 65 |
prompt = f"[INST] Write SQLite query to answer the following question given the database schema. Please wrap your code answer using ```: Schema: {table} Question: {question} [/INST] Here is the SQLite query to answer to the question: {question}: ``` "
|
| 66 |
|
| 67 |
output = model(prompt)
|
| 68 |
+
output = output.split('```')[2]
|
| 69 |
print(output)
|
| 70 |
```
|
| 71 |
|