Update README.md

README.md

@@ -52,7 +52,6 @@ prompt = f"Instruct: generate a SQL query.\n{prompt}\nOutput:\n" # for dataset b
 outputs = pipe(prompt, max_new_tokens=1024, do_sample=True, temperature=0.9, top_k=50, top_p=0.1, eos_token_id=pipe.tokenizer.eos_token_id, pad_token_id=pipe.tokenizer.eos_token_id)
 
 print('Question: %s'%prompt)
-#print('Answer: %s \nOutput:\n'%outputs[0]['generated_text'])
 print(f"Generated Answer:\n{outputs[0]['generated_text'][len(prompt):].strip()}")
 
 Question: Instruct: generate a SQL query.
@@ -69,7 +68,6 @@ prompt = f"Instruct: Answer the following question.\n{prompt}\nOutput:\n"
 outputs = pipe(prompt, max_new_tokens=1024, do_sample=True, temperature=0.9, top_k=50, top_p=0.1, eos_token_id=pipe.tokenizer.eos_token_id, pad_token_id=pipe.tokenizer.eos_token_id)
 
 print('Question: %s'%prompt)
-#print('Answer: %s \nOutput:\n'%outputs[0]['generated_text'])
 print(f"Generated Answer:\n{outputs[0]['generated_text'][len(prompt):].strip()}")
 
 Question: Instruct: Answer the following question.
@@ -79,6 +77,25 @@ Output:
 Generated Answer:
 The first album Beyoncé released as a solo artist was "Dangerously in Love".
 
+
+CASE Number 3:
+prompt='What was the first album Beyoncé released as a solo artist?'
+prompt = f"Instruct: generate a SQL query.\n{prompt}\n\n" # for dataset b-mc2/sql-create-context
+outputs = pipe(prompt, max_new_tokens=1024, do_sample=True, temperature=0.9, top_k=50, top_p=0.1, eos_token_id=pipe.tokenizer.eos_token_id, pad_token_id=pipe.tokenizer.eos_token_id)
+
+print('Question: %s'%prompt)
+print(f"Generated Answer:\n{outputs[0]['generated_text'][len(prompt):].strip()}")
+
+
+Question: Instruct: generate a SQL query.
+What was the first album Beyoncé released as a solo artist?
+
+
+Generated Answer:
+```sql
+SELECT first_album FROM table_name_84 WHERE solo_artist = "beyoncé"
+
+
 ## Intended uses & limitations
 
 More information needed