from transformers import AutoModelForSemanticSegmentation, AutoTokenizer
from PIL import Image
import requests
from io import BytesIO

# Hugging Face 모델 이름
model_name = "nvidia/segformer-b0-finetuned-cityscapes-1024-1024"

# 모델 및 토크나이저 로드
model = AutoModelForSemanticSegmentation.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)

# 이미지 다운로드 및 전처리
image_url = "https://example.com/your_image.jpg"  # 이미지 URL을 실제 이미지 URL로 바꿔주세요.
image = Image.open(BytesIO(requests.get(image_url).content))

# 이미지를 모델 입력 형식으로 변환
inputs = tokenizer(image, return_tensors="pt")

# 모델 실행
outputs = model(**inputs)
predictions = outputs.logits

# 여기서 predictions는 segmentation 맵입니다.
# 이를 원하는 형식으로 처리하고 시각화할 수 있습니다.
# 예를 들어, PIL Image로 변환하여 저장하거나 출력할 수 있습니다.