Update README.md

README.md

@@ -25,7 +25,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("ekshat/Llama-2-7b-chat-finetune-for-text2sql")
 model = AutoModelForCausalLM.from_pretrained("ekshat/Llama-2-7b-chat-finetune-for-text2sql")
 
-# Run text generation pipeline with our next model
+# Run text generation pipeline with our model
 context = "CREATE TABLE Student (name VARCHAR, college VARCHAR, age VARCHAR, group VARCHAR, marks VARCHAR)"
 question = "List the name of Students belongs to school 'St. Xavier' and having marks greater than '600'"
 
@@ -37,14 +37,14 @@ prompt = f"""Below is an context that describes a sql query, paired with an ques
 ### Answer:"""
 
 sequences = pipeline(
-prompt,
-do_sample=True,
-top_k=10,
-num_return_sequences=1,
-eos_token_id=tokenizer.eos_token_id,
-max_length=200,
+                      prompt,
+                      do_sample=True,
+                      top_k=10,
+                      num_return_sequences=1,
+                      eos_token_id=tokenizer.eos_token_id,
+                      max_length=200,
 )
 for seq in sequences:
-print(f"Result: {seq['generated_text']}")
+  print(f"Result: {seq['generated_text']}")
 
 ```