Fix zero-shot segmentation section, use public example image

README.md

@@ -24,12 +24,15 @@ pip install transformers torch torchvision sentencepiece
 from transformers import AutoModel
 from torchvision import transforms
 from PIL import Image
+import requests
 
 model = AutoModel.from_pretrained("google/tipsv2-g14-dpt", trust_remote_code=True)
 model.eval().cuda()
 
+url = "https://raw.githubusercontent.com/google-deepmind/tips/main/scenic/images/example_image.jpg"
+image = Image.open(requests.get(url, stream=True).raw)
 transform = transforms.Compose([transforms.Resize((448, 448)), transforms.ToTensor()])
-pixel_values = transform(Image.open("photo.jpg")).unsqueeze(0).cuda()
+pixel_values = transform(image).unsqueeze(0).cuda()
 
 # All tasks at once
 outputs = model(pixel_values)