Update README.md

README.md

@@ -44,4 +44,18 @@ def get_sql(query,tokenizer,model):
 query="Show me the average age of of wines in Italy by provinces"
 sql = get_sql(query,tokenizer,model)
 print(sql)
+
+#https://huggingface.co/mrm8488/t5-small-finetuned-wikiSQL
+def get_sql(query):
+  input_text = "translate English to SQL: %s </s>" % query
+  features = tokenizer([input_text], return_tensors='pt')
+
+  output = model.generate(input_ids=features['input_ids'], 
+               attention_mask=features['attention_mask'])
+  
+  return tokenizer.decode(output[0])
+
+query = "How many models were finetuned using BERT as base model?"
+
+get_sql(query)
 ```