| | --- |
| | language: en |
| | tags: |
| | - text2sql |
| | - causal-lm |
| | - transformers |
| | - safetensors |
| | license: mit |
| | pipeline_tag: text-generation |
| | --- |
| | |
| |
|
| |
|
| | # Phi-3 Text-to-SQL Model |
| |
|
| | This is a fine-tuned **Microsoft Phi-3** model specialized for **Text-to-SQL** generation. |
| |
|
| | ## Example |
| |
|
| | ```python |
| | from transformers import AutoModelForCausalLM, AutoTokenizer |
| | |
| | repo_id = "bhavika24/text2sql" |
| | tokenizer = AutoTokenizer.from_pretrained(repo_id) |
| | model = AutoModelForCausalLM.from_pretrained(repo_id, torch_dtype="auto", device_map="auto") |
| | |
| | question = "List all customers who ordered products over $500 last month." |
| | inputs = tokenizer(question, return_tensors="pt").to(model.device) |
| | outputs = model.generate(**inputs, max_new_tokens=128) |
| | |
| | sql_query = tokenizer.decode(outputs[0], skip_special_tokens=True) |
| | print(sql_query) |
| | |
