ansumanpandey
/

codgen-finetuned-SQLQueryGeneration

Text Generation

code-generation

finetuned-model

Model card Files Files and versions

ansumanpandey commited on Apr 15, 2024

Commit

2dcc55d

·

verified ·

1 Parent(s): b44bdcb

Update The SQL Generation Code

Files changed (1) hide show

The SQL Generation Code +1 -1

The SQL Generation Code CHANGED Viewed

@@ -4,7 +4,7 @@ tokenizer = AutoTokenizer.from_pretrained("ansumanpandey/codgen-finetuned-SQLQue
 model = AutoModelForCausalLM.from_pretrained("ansumanpandey/codgen-finetuned-SQLQueryGeneration")
 def get_sql(query):
-  input_text = "Write a SQL query to %s </s>" % query
   features = tokenizer([input_text], return_tensors='pt')
   output = model.generate(input_ids=features['input_ids'],

 model = AutoModelForCausalLM.from_pretrained("ansumanpandey/codgen-finetuned-SQLQueryGeneration")
 def get_sql(query):
+  input_text = "Query to %s </s>" % query
   features = tokenizer([input_text], return_tensors='pt')
   output = model.generate(input_ids=features['input_ids'],