ansumanpandey
commited on
Commit
•
7c85e3b
1
Parent(s):
2c5b451
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 |
+
***************************
|