Update README.md
Browse files
README.md
CHANGED
@@ -21,7 +21,7 @@ model.cuda()
|
|
21 |
config = model.config
|
22 |
tokenizer = AutoTokenizer.from_pretrained(hf_path, use_fast=False, model_max_length = config.tokenizer_model_max_length,padding_side = config.tokenizer_padding_side)
|
23 |
prompt="What are these?"
|
24 |
-
image_url="http://images.cocodataset.org/
|
25 |
output_text, genertaion_time = model.chat(prompt=prompt, image=image_url, tokenizer=tokenizer)
|
26 |
|
27 |
print('model output:', output_text)
|
|
|
21 |
config = model.config
|
22 |
tokenizer = AutoTokenizer.from_pretrained(hf_path, use_fast=False, model_max_length = config.tokenizer_model_max_length,padding_side = config.tokenizer_padding_side)
|
23 |
prompt="What are these?"
|
24 |
+
image_url="http://images.cocodataset.org/test-stuff2017/000000000001.jpg"
|
25 |
output_text, genertaion_time = model.chat(prompt=prompt, image=image_url, tokenizer=tokenizer)
|
26 |
|
27 |
print('model output:', output_text)
|