Hugging Face's logo

ansumanpandey
/
codgen-finetuned-SQLQueryGeneration

Text Generation
Transformers
Safetensors
English
codegen
nlp
code-generation
sql
finetuned-model
Inference Endpoints
Model card Files Community
ansumanpandey commited on
Commit
2dcc55d
·
verified ·
1 Parent(s): b44bdcb

Update The SQL Generation Code

Browse files
Files changed (1) hide show
  1. The SQL Generation Code +1 -1
The SQL Generation Code CHANGED
@@ -4,7 +4,7 @@ tokenizer = AutoTokenizer.from_pretrained("ansumanpandey/codgen-finetuned-SQLQue
4
  model = AutoModelForCausalLM.from_pretrained("ansumanpandey/codgen-finetuned-SQLQueryGeneration")
5
 
6
  def get_sql(query):
7
- input_text = "Write a SQL query to %s </s>" % query
8
  features = tokenizer([input_text], return_tensors='pt')
9
 
10
  output = model.generate(input_ids=features['input_ids'],
 
4
  model = AutoModelForCausalLM.from_pretrained("ansumanpandey/codgen-finetuned-SQLQueryGeneration")
5
 
6
  def get_sql(query):
7
+ input_text = "Query to %s </s>" % query
8
  features = tokenizer([input_text], return_tensors='pt')
9
 
10
  output = model.generate(input_ids=features['input_ids'],