osllmai
/

text-to-sql

text2text-generation

natural-language-to-sql

text-generation-inference

Model card Files Files and versions Community

fghcol30 commited on Feb 18

Commit

c09153a

·

verified ·

1 Parent(s): 6eaaf53

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -44,8 +44,8 @@ The model may generate incorrect or unsafe SQL queries if the input question is
 from transformers import T5Tokenizer, T5ForConditionalGeneration
 # Load the fine-tuned model
-model = T5ForConditionalGeneration.from_pretrained("fghcol30/text-to-sql")
-tokenizer = T5Tokenizer.from_pretrained("fghcol30/text-to-sql")
 # Generate SQL query
 def generate_sql_query(question):

 from transformers import T5Tokenizer, T5ForConditionalGeneration
 # Load the fine-tuned model
+model = T5ForConditionalGeneration.from_pretrained("osllmai/text-to-sql")
+tokenizer = T5Tokenizer.from_pretrained("osllmai/text-to-sql")
 # Generate SQL query
 def generate_sql_query(question):