Hugging Face's logo

osllmai
/
text-to-sql

Transformers
Safetensors
t5
text2text-generation
text2sql
natural-language-to-sql
spider-dataset
text-generation-inference
Model card Files
xet
 Community
fghcol30 commited on
Commit
c09153a
·
verified ·
1 Parent(s): 6eaaf53

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -44,8 +44,8 @@ The model may generate incorrect or unsafe SQL queries if the input question is
44
  from transformers import T5Tokenizer, T5ForConditionalGeneration
45
 
46
  # Load the fine-tuned model
47
- model = T5ForConditionalGeneration.from_pretrained("fghcol30/text-to-sql")
48
- tokenizer = T5Tokenizer.from_pretrained("fghcol30/text-to-sql")
49
 
50
  # Generate SQL query
51
  def generate_sql_query(question):
 
44
  from transformers import T5Tokenizer, T5ForConditionalGeneration
45
 
46
  # Load the fine-tuned model
47
+ model = T5ForConditionalGeneration.from_pretrained("osllmai/text-to-sql")
48
+ tokenizer = T5Tokenizer.from_pretrained("osllmai/text-to-sql")
49
 
50
  # Generate SQL query
51
  def generate_sql_query(question):