update readme

README.md

@@ -44,7 +44,7 @@ class EosListStoppingCriteria(StoppingCriteria):
 # load models
 model_name_or_path = "Salesforce/blip3-phi3-3b-instruct-r-v1"
 model = AutoModelForVision2Seq.from_pretrained(model_name_or_path, trust_remote_code=True)
-tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, trust_remote_code=True, use_fast=True, legacy=False)
+tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, trust_remote_code=True, use_fast=False, legacy=False)
 image_processor = AutoImageProcessor.from_pretrained(model_name_or_path, trust_remote_code=True)
 tokenizer = model.update_special_tokens(tokenizer)
 
@@ -64,7 +64,7 @@ generated_text = model.generate(**inputs, image_size=[raw_image.size],
                                 do_sample=False, max_new_tokens=768, top_p=None, num_beams=1,
                                 stopping_criteria = [EosListStoppingCriteria()],
                                 )
-prediction = tokenizer.decode(generated_text[0], skip_special_tokens=True)
+prediction = tokenizer.decode(generated_text[0], skip_special_tokens=True).split("<|end|>")[0]
 print("==> prediciton: ", prediction)
 # output: ==> prediciton: There is one dog in the picture.
 ```