added colab

README.md

@@ -16,6 +16,8 @@ arxiv: 2408.03326
 
 ![image/png](llava_onevision_arch.png)
 
+Check out also the Google Colab demo to run Llava on a free-tier Google Colab instance: [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1-4AtYjR8UMtCALV0AswU1kiNkWCLTALT?usp=sharing)
+
 Below is the model card of 0.5B LLaVA-Onevision model which is copied from the original LLaVA-Onevision model card that you can find [here](https://huggingface.co/lmms-lab/llava-onevision-qwen2-0.5b-si).
 
 
@@ -56,8 +58,11 @@ Below we used [`"llava-hf/llava-onevision-qwen2-0.5b-ov-hf"`](https://huggingfac
 from transformers import pipeline
 from PIL import Image    
 import requests
+from transformers import AutoProcessor
+
 
 model_id = "llava-hf/llava-onevision-qwen2-0.5b-ov-hf"
+processor = AutoProcessor.from_pretrained(model_id)
 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)