llava-hf
/

bakLlava-v1-hf

Image-Text-to-Text

Inference Endpoints

Model card Files Files and versions Community

ybelkada commited on Dec 6, 2023

Commit

d47ad64

·

1 Parent(s): 9fd56c0

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -39,7 +39,7 @@ pipe = pipeline("image-to-text", model=model_id)
 url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/ai2d-demo.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
-prompt = "<image>\nUSER: What does the label 15 represent? (1) lava (2) core (3) tunnel (4) ash cloud\nASSISTANT:"
 outputs = pipe(image, prompt=prompt, generate_kwargs={"max_new_tokens": 200})
 print(outputs)
@@ -59,7 +59,7 @@ from transformers import AutoProcessor, LlavaForConditionalGeneration
 model_id = "llava-hf/bakLlava-v1-hf"
-prompt = "<image> \nUSER: What are these?\nASSISTANT:"
 image_file = "http://images.cocodataset.org/val2017/000000039769.jpg"
 model = LlavaForConditionalGeneration.from_pretrained(

 url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/ai2d-demo.jpg"
 image = Image.open(requests.get(url, stream=True).raw)
+prompt = "USER: <image>\nWhat does the label 15 represent? (1) lava (2) core (3) tunnel (4) ash cloud\nASSISTANT:"
 outputs = pipe(image, prompt=prompt, generate_kwargs={"max_new_tokens": 200})
 print(outputs)
 model_id = "llava-hf/bakLlava-v1-hf"
+prompt = "USER: <image>\nWhat are these?\nASSISTANT:"
 image_file = "http://images.cocodataset.org/val2017/000000039769.jpg"
 model = LlavaForConditionalGeneration.from_pretrained(