Spaces:
Build error
Build error
Commit
·
97fb1b5
1
Parent(s):
70c26df
Update app.py
Browse files
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"
|
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
|