Spaces:
Runtime error
Runtime error
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 | |