ansumanpandey
/

codgen-finetuned-SQLQueryGeneration

Text Generation

code-generation

finetuned-model

Inference Endpoints

Model card Files Files and versions Community

ansumanpandey commited on Apr 15

Commit

2dcc55d

•

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'],