NGrov commited on
Commit
b347b07
1 Parent(s): cbda07a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -101,7 +101,7 @@ input_text = " ".join(["Question: ",question, "Schema:", schema])
101
  model_inputs = tokenizer(input_text, return_tensors="pt")
102
  outputs = model.generate(**model_inputs, max_length=512)
103
 
104
- output_text = tokenizer.decode(outputs, skip_special_tokens=True)
105
 
106
  print("SQL Query:")
107
  print(output_text)
 
101
  model_inputs = tokenizer(input_text, return_tensors="pt")
102
  outputs = model.generate(**model_inputs, max_length=512)
103
 
104
+ output_text = tokenizer.batch_decode(outputs, skip_special_tokens=True)
105
 
106
  print("SQL Query:")
107
  print(output_text)