Sid26Roy commited on
Commit
9f67492
·
verified ·
1 Parent(s): a0ff27a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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