Update app.py
Browse files
app.py
CHANGED
@@ -33,7 +33,7 @@ SQL Query:"""
|
|
33 |
|
34 |
inputs = tokenizer(prompt, return_tensors="pt", truncation=True, padding=True).to(device)
|
35 |
with torch.no_grad():
|
36 |
-
outputs = model.generate(**inputs, max_length=256)
|
37 |
generated_sql = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
38 |
return generated_sql
|
39 |
|
|
|
33 |
|
34 |
inputs = tokenizer(prompt, return_tensors="pt", truncation=True, padding=True).to(device)
|
35 |
with torch.no_grad():
|
36 |
+
outputs = model.generate(**inputs, max_length=256 , decoder_start_token_id=tokenizer.convert_tokens_to_ids(tokenizer.pad_token))
|
37 |
generated_sql = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
38 |
return generated_sql
|
39 |
|