Update README.md
Browse files
README.md
CHANGED
|
@@ -29,6 +29,8 @@ This model is intended for the automatic generation of SQL queries from descript
|
|
| 29 |
|
| 30 |
This model can be used with the Hugging Face Transformers library. Here is a quick example:
|
| 31 |
|
|
|
|
|
|
|
| 32 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 33 |
|
| 34 |
tokenizer = AutoTokenizer.from_pretrained("ansumanpandey/codgen-finetuned-SQLQueryGeneration")
|
|
@@ -42,4 +44,6 @@ def get_sql(query):
|
|
| 42 |
attention_mask=features['attention_mask'],
|
| 43 |
max_new_tokens=70)
|
| 44 |
sql_query= tokenizer.decode(output[0])
|
| 45 |
-
return sql_query
|
|
|
|
|
|
|
|
|
| 29 |
|
| 30 |
This model can be used with the Hugging Face Transformers library. Here is a quick example:
|
| 31 |
|
| 32 |
+
***************************
|
| 33 |
+
|
| 34 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 35 |
|
| 36 |
tokenizer = AutoTokenizer.from_pretrained("ansumanpandey/codgen-finetuned-SQLQueryGeneration")
|
|
|
|
| 44 |
attention_mask=features['attention_mask'],
|
| 45 |
max_new_tokens=70)
|
| 46 |
sql_query= tokenizer.decode(output[0])
|
| 47 |
+
return sql_query
|
| 48 |
+
|
| 49 |
+
***************************
|