Update README.md

README.md

@@ -18,11 +18,8 @@ Our Model is fine tuned on Llama-2 7B model on text-2-sql Dataset on alpaca form
 
 !pip install transformers accelerate
 
-from transformers import pipeline
-pipe = pipeline("text-generation", model="ekshat/Llama-2-7b-chat-finetune-for-text2sql")
-
 # Load model directly
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
 tokenizer = AutoTokenizer.from_pretrained("ekshat/Llama-2-7b-chat-finetune-for-text2sql")
 model = AutoModelForCausalLM.from_pretrained("ekshat/Llama-2-7b-chat-finetune-for-text2sql")