From 7993b42804709ce3274ba2bfb327e98ca0dc2e7b Mon Sep 17 00:00:00 2001 From: Niels Rogge Date: Mon, 21 Feb 2022 20:06:00 +0000 Subject: [PATCH] Update code example --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 331f32e..190af00 100644 --- a/README.md +++ b/README.md @@ -35,8 +35,8 @@ from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmenta from PIL import Image import requests -feature_extractor = SegformerFeatureExtractor.from_pretrained("nvidia/segformer-b0-finetuned-ade-512-512") -model = SegformerForSemanticSegmentation.from_pretrained("nvidia/segformer-b0-finetuned-ade-512-512") +feature_extractor = SegformerFeatureExtractor.from_pretrained("nvidia/segformer-b5-finetuned-ade-512-512") +model = SegformerForSemanticSegmentation.from_pretrained("nvidia/segformer-b5-finetuned-ade-512-512") url = "http://images.cocodataset.org/val2017/000000039769.jpg" image = Image.open(requests.get(url, stream=True).raw)