Update README.md

README.md

@@ -48,10 +48,13 @@ pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 CASE Number 1:
 
 prompt='What was the first album Beyoncé released as a solo artist?'
+
 prompt = f"Instruct: generate a SQL query.\n{prompt}\nOutput:\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.
@@ -64,10 +67,13 @@ SELECT first_album FROM table_name_82 WHERE solo_artist = "beyoncé"
 CASE Number 2:
 
 prompt='What was the first album Beyoncé released as a solo artist?'
+
 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(f"Generated Answer:\n{outputs[0]['generated_text'][len(prompt):].strip()}")
 
 Question: Instruct: Answer the following question.