Views
No views yet
t5-base)t5-base on a custom Text-to-SQL dataset. It translates natural language questions into corresponding SQL queries.1from transformers import T5Tokenizer, T5ForConditionalGeneration
2
3model = T5ForConditionalGeneration.from_pretrained("Priyanshu05/text-to-sql-t5")
4tokenizer = T5Tokenizer.from_pretrained("Priyanshu05/text-to-sql-t5")
5
6question = "translate natural language to SQL: show all customers"
7inputs = tokenizer(question, return_tensors="pt")
8output = model.generate(**inputs)
9print(tokenizer.decode(output[0], skip_special_tokens=True))