blacknahil/t5-base-spider-v1-finetuned

T5-Base Spider (Robust Version)

This model is fine-tuned on the Spider dataset to translate natural language questions into SQL queries.

Model Details

• Base Model: google/flan-t5-base
• Dataset: Spider (Text-to-SQL)
• Training Epochs: 3
• Training Time: ~1:30Hours

How to Use

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

tokenizer = AutoTokenizer.from_pretrained("blacknahil/t5-base-spider-v1-finetuned")
model = AutoModelForSeq2SeqLM.from_pretrained("blacknahil/t5-base-spider-v1-finetuned")

input_text = "Translate to SQL: Show all students who are older than 20 | schema: student : id , name , age"
inputs = tokenizer(input_text, return_tensors="pt")
outputs = model.generate(**inputs, max_length=128)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))