Update README.md
Browse files
README.md
CHANGED
|
@@ -19,7 +19,7 @@ You can then use this model to translate SQL queries into plain english.
|
|
| 19 |
|
| 20 |
```python
|
| 21 |
query = "SELECT COUNT Params from model where location=HF-Hub"
|
| 22 |
-
input_text = f"translate
|
| 23 |
features = tokenizer([input_text], return_tensors='pt')
|
| 24 |
|
| 25 |
output = model.generate(input_ids=features['input_ids'].cuda(),
|
|
|
|
| 19 |
|
| 20 |
```python
|
| 21 |
query = "SELECT COUNT Params from model where location=HF-Hub"
|
| 22 |
+
input_text = f"translate SQL to English: {query} </s>"
|
| 23 |
features = tokenizer([input_text], return_tensors='pt')
|
| 24 |
|
| 25 |
output = model.generate(input_ids=features['input_ids'].cuda(),
|