--- title: Machine Learning 7task 2 emoji: 🌖 colorFrom: purple colorTo: indigo sdk: gradio sdk_version: 3.44.4 app_file: app.py pinned: false --- ```python from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmentation from PIL import Image import requests feature_extractor = SegformerFeatureExtractor.from_pretrained("nvidia/segformer-b3-finetuned-cityscapes-1024-1024") model = SegformerForSemanticSegmentation.from_pretrained("nvidia/segformer-b3-finetuned-cityscapes-1024-1024") url = ["https://imagescdn.gettyimagesbank.com/500/201407/a8281860.jpg", "https://imagescdn.gettyimagesbank.com/500/14/890/614/3/a8281872.jpg", "https://50plus.or.kr/upload/im/2019/06/d3007817-9c63-468d-bbc1-0afa8f10874e.jpg" ] image = Image.open(requests.get(url, stream=True).raw) inputs = feature_extractor(images=image, return_tensors="pt") outputs = model(**inputs) logits = outputs.logits # shape (batch_size, num_labels, height/4, width/4) ``` Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference