Update README.md

README.md

@@ -18,9 +18,11 @@ or  execute the following test code:
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from generate_model import *
-model = AutoModelForCausalLM.from_pretrained("tinyllava/TinyLLaVA-Phi-2-SigLIP-3.1B", trust_remote_code=True)
+
+hf_path = 'tinyllava/TinyLLaVA-Phi-2-SigLIP-3.1B'
+model = AutoModelForCausalLM.from_pretrained(hf_path, trust_remote_code=True)
 config = model.config
-tokenizer = AutoTokenizer.from_pretrained("tinyllava/TinyLLaVA-Phi-2-SigLIP-3.1B", use_fast=False, model_max_length = config.tokenizer_model_max_length,padding_side = config.tokenizer_padding_side)
+tokenizer = AutoTokenizer.from_pretrained(hf_path, use_fast=False, model_max_length = config.tokenizer_model_max_length,padding_side = config.tokenizer_padding_side)
 prompt="you want to ask"
 image="/path/to/related/image"
 output_text, genertaion_time = generate(prompt=prompt, image=image, model=model, tokenizer=tokenizer)