RealMati commited on
Commit
5132d2c
·
verified ·
1 Parent(s): 963ce31

Add model card with pipeline_tag for Inference API

Browse files
Files changed (1) hide show
  1. README.md +44 -0
README.md ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ pipeline_tag: text2text-generation
3
+ library_name: transformers
4
+ tags:
5
+ - t5
6
+ - text2text-generation
7
+ - sql
8
+ - text-to-sql
9
+ - wikisql
10
+ ---
11
+
12
+ # T2SQL V6 Structured - Text to SQL
13
+
14
+ Fine-tuned T5 model that converts natural language questions to SQL queries.
15
+
16
+ ## Usage
17
+
18
+ ```python
19
+ from transformers import pipeline
20
+
21
+ pipe = pipeline("text2text-generation", model="RealMati/t2sql_v6_structured")
22
+ result = pipe("translate to SQL: list all users older than 18 | schema: users(id, name, age, email)")
23
+ print(result[0]["generated_text"])
24
+ ```
25
+
26
+ ## Inference API
27
+
28
+ ```python
29
+ import requests
30
+
31
+ API_URL = "https://router.huggingface.co/hf-inference/models/RealMati/t2sql_v6_structured"
32
+ headers = {"Authorization": "Bearer hf_YOUR_TOKEN"}
33
+
34
+ response = requests.post(API_URL, headers=headers, json={
35
+ "inputs": "translate to SQL: How many users are there? | schema: users(id, name, age)"
36
+ })
37
+ print(response.json())
38
+ ```
39
+
40
+ ## Training
41
+
42
+ - **Base model:** T5-base
43
+ - **Dataset:** WikiSQL (56k train / 8k val / 15k test)
44
+ - **Task:** Natural language to structured SQL output