AjayP13 commited on
Commit
740ccf5
·
verified ·
1 Parent(s): 98234bc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -20,10 +20,10 @@ The output will be a JSON object.
20
  ```python
21
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline, Conversation
22
 
23
- # Load model
24
- tokenizer = AutoTokenizer.from_pretrained('fineinstructions/query_templatizer', revision=None) # Load tokenizer
25
  tokenizer.padding_side = 'left'
26
- model = AutoModelForCausalLM.from_pretrained('fineinstructions/query_templatizer', revision=None) # Load model
27
  pipe = pipeline('text-generation', model=model, tokenizer=tokenizer, pad_token_id=tokenizer.pad_token_id, return_full_text=False)
28
 
29
  # Run inference to templatize the query
 
20
  ```python
21
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline, Conversation
22
 
23
+ # Load tokenizer and model
24
+ tokenizer = AutoTokenizer.from_pretrained('fineinstructions/query_templatizer', revision=None)
25
  tokenizer.padding_side = 'left'
26
+ model = AutoModelForCausalLM.from_pretrained('fineinstructions/query_templatizer', revision=None)
27
  pipe = pipeline('text-generation', model=model, tokenizer=tokenizer, pad_token_id=tokenizer.pad_token_id, return_full_text=False)
28
 
29
  # Run inference to templatize the query