Update README.md

README.md

@@ -69,19 +69,14 @@ Model inference
 question = "server of user id 11 with status active and server id 10"
 table = "table_name_77 ( user id INTEGER, status VARCHAR, server id INTEGER )"
 
-
-
-
-
 test = f"""Question: {question}\nTable: {table}\nSQL: """
 
 p = tokenizer(test, return_tensors='pt')
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
+out = model.to(device).generate(**p.to(device),max_new_tokens=50)
 
-
-
-print("output :", tokenizer.batch_decode(model.to(device).generate(**p.to(device),max_new_tokens=50),skip_special_tokens=True)[0])
+print("SQL Query :", tokenizer.batch_decode(out,skip_special_tokens=True)[0])
 
 ```