Update README.md

README.md

@@ -17,7 +17,7 @@ See the list below for the details of each model:
 2. running the following command:
 
 ```bash
-python generate_model --model [MODEL_NAME] --prompt 'you want to ask' --image '/path/to/related/image'
+python generate_model --model jiajunlong/TinyLLaVA-0.89B --prompt 'you want to ask' --image '/path/to/related/image'
 ```
 
 or  execute the following test code:
@@ -25,11 +25,11 @@ or  execute the following test code:
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from generate_model import *
-model = AutoModelForCausalLM.from_pretrained("jiajunlong/TinyLLaVA-0.55B", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("jiajunlong/TinyLLaVA-0.89B", trust_remote_code=True)
 config = model.config
-tokenizer = AutoTokenizer.from_pretrained("jiajunlong/TinyLLaVA-0.55B", use_fast=False, model_max_length = config.tokenizer_model_max_length,padding_side = config.tokenizer_padding_side)
-prompt="what is umusual about this image?"
-image="joke.png"
+tokenizer = AutoTokenizer.from_pretrained("jiajunlong/TinyLLaVA-0.89B", 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)
 print_txt = (
         f'\r\n{"=" * os.get_terminal_size().columns}\r\n'