Fix inference script image download and formatting

README.md

@@ -136,6 +136,7 @@ Then you can run the following code:
 ```python
 import torch
 from PIL import Image
+from io import BytesIO
 import requests
 
 from transformers import AutoModelForCausalLM, AutoProcessor
@@ -146,11 +147,11 @@ processor = AutoProcessor.from_pretrained("microsoft/Magma-8B", trust_remote_cod
 model.to("cuda")
 
 # Inference
-url = "https://assets-c4akfrf5b4d3f4b7.z01.azurefd.net/assets/2024/04/BMDataViz_661fb89f3845e.png" 
-image = Image.open(requests.get(url, stream=True).raw) 
+url = "https://assets-c4akfrf5b4d3f4b7.z01.azurefd.net/assets/2024/04/BMDataViz_661fb89f3845e.png"
+image = Image.open(BytesIO(requests.get(url, stream=True).content))
 
 convs = [
-    {"role": "system", "content": "You are agent that can see, talk and act."},            
+    {"role": "system", "content": "You are agent that can see, talk and act."},
     {"role": "user", "content": "<image_start><image><image_end>\nWhat is in this image?"},
 ]
 prompt = processor.tokenizer.apply_chat_template(convs, tokenize=False, add_generation_prompt=True)