t2sql_v6_structured / README.md
RealMati's picture
Fix pipeline_tag to text-generation
d7473da verified
metadata
pipeline_tag: text-generation
library_name: transformers
tags:
  - t5
  - text2text-generation
  - sql
  - text-to-sql
  - wikisql

T2SQL V6 Structured - Text to SQL

Fine-tuned T5 model that converts natural language questions to SQL queries.

Usage

from transformers import pipeline

pipe = pipeline("text2text-generation", model="RealMati/t2sql_v6_structured")
result = pipe("translate to SQL: list all users older than 18 | schema: users(id, name, age, email)")
print(result[0]["generated_text"])

Training

  • Base model: T5-base
  • Dataset: WikiSQL (56k train / 8k val / 15k test)
  • Task: Natural language to structured SQL output