beingpraveen commited on
Commit
97fb1b5
·
1 Parent(s): 70c26df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ model = layer.get_model('layer/t5-fine-tuning-with-layer/models/t5-english-to-sq
5
  tokenizer = layer.get_model('layer/t5-fine-tuning-with-layer/models/t5-tokenizer').get_train()
6
 
7
  def convert(query):
8
- inputs = tokenizer.encode(f"text to SQL: {query}", return_tensors="pt")
9
  outputs = model.generate(inputs, max_length=1024)
10
  sql = tokenizer.decode(outputs[0], skip_special_tokens=True)
11
  return sql
 
5
  tokenizer = layer.get_model('layer/t5-fine-tuning-with-layer/models/t5-tokenizer').get_train()
6
 
7
  def convert(query):
8
+ inputs = tokenizer.encode(f"translate English to SQL: {query}", return_tensors="pt")
9
  outputs = model.generate(inputs, max_length=1024)
10
  sql = tokenizer.decode(outputs[0], skip_special_tokens=True)
11
  return sql