Update inference.py
This commit is contained in:
parent
c167d340ad
commit
243102a264
|
@ -48,8 +48,7 @@ def run_inference(question, prompt_file="prompt.md", metadata_file="metadata.sql
|
|||
eos_token_id=eos_token_id,
|
||||
pad_token_id=eos_token_id,
|
||||
)[0]["generated_text"]
|
||||
.split("# SQL\n")[-1]
|
||||
.split("```")[0]
|
||||
.split("```")[1]
|
||||
.split(";")[0]
|
||||
.strip()
|
||||
+ ";"
|
||||
|
|
Loading…
Reference in New Issue