Update README.md

README.md

@@ -20,10 +20,10 @@ The output will be a JSON object.
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline, Conversation
 
-# Load model
-tokenizer = AutoTokenizer.from_pretrained('fineinstructions/query_templatizer', revision=None) # Load tokenizer
+# Load tokenizer and model
+tokenizer = AutoTokenizer.from_pretrained('fineinstructions/query_templatizer', revision=None)
 tokenizer.padding_side = 'left'
-model = AutoModelForCausalLM.from_pretrained('fineinstructions/query_templatizer', revision=None) # Load model
+model = AutoModelForCausalLM.from_pretrained('fineinstructions/query_templatizer', revision=None)
 pipe = pipeline('text-generation', model=model, tokenizer=tokenizer, pad_token_id=tokenizer.pad_token_id, return_full_text=False)
 
 # Run inference to templatize the query