Update code example

README.md

@@ -31,13 +31,13 @@ Here is how to use this model for semantic segmentation:
 
 ```python
 from transformers import BeitFeatureExtractor, BeitForSemanticSegmentation
+from datasets import load_dataset
 from PIL import Image
-import requests
 
-url = 'http://images.cocodataset.org/val2017/000000039769.jpg'
-image = Image.open(requests.get(url, stream=True).raw)
+# load ADE20k image
+ds = load_dataset("hf-internal-testing/fixtures_ade20k", split="test")
 
-feature_extractor = BeitFeatureExtractor(do_resize=True, size=640, do_center_crop=False)
+feature_extractor = BeitFeatureExtractor.from_pretrained('microsoft/beit-base-finetuned-ade-640-640')
 model = BeitForImageClassification.from_pretrained('microsoft/beit-base-finetuned-ade-640-640')
 
 inputs = feature_extractor(images=image, return_tensors="pt")