ansumanpandey commited on
Commit
7c85e3b
1 Parent(s): 2c5b451

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
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
+ ***************************