dsivakumar commited on
Commit
ee84283
·
1 Parent(s): 73a545c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -0
README.md CHANGED
@@ -1,14 +1,17 @@
1
  #import transformers
 
2
  from transformers import (
3
  T5ForConditionalGeneration,
4
  T5Tokenizer,
5
  )
6
 
7
  #load model
 
8
  model = T5ForConditionalGeneration.from_pretrained('dsivakumar/text2sql')
9
  tokenizer = T5Tokenizer.from_pretrained('dsivakumar/text2sql')
10
 
11
  #predict function
 
12
  def get_sql(query,tokenizer,model):
13
  source_text= "English to SQL: "+query
14
  source_text = ' '.join(source_text.split())
@@ -28,6 +31,8 @@ def get_sql(query,tokenizer,model):
28
  return preds
29
 
30
  #test
 
31
  query="Show me the average age of of wines in Italy by provinces"
32
  sql = get_sql(query,tokenizer,model)
33
  print(sql)
 
 
1
  #import transformers
2
+
3
  from transformers import (
4
  T5ForConditionalGeneration,
5
  T5Tokenizer,
6
  )
7
 
8
  #load model
9
+
10
  model = T5ForConditionalGeneration.from_pretrained('dsivakumar/text2sql')
11
  tokenizer = T5Tokenizer.from_pretrained('dsivakumar/text2sql')
12
 
13
  #predict function
14
+
15
  def get_sql(query,tokenizer,model):
16
  source_text= "English to SQL: "+query
17
  source_text = ' '.join(source_text.split())
 
31
  return preds
32
 
33
  #test
34
+
35
  query="Show me the average age of of wines in Italy by provinces"
36
  sql = get_sql(query,tokenizer,model)
37
  print(sql)
38
+