1

app.py

@@ -4,13 +4,13 @@ from PIL import Image
 import torch
 
 # 모델과 feature extractor 로드
-model = SegformerForSemanticSegmentation.from_pretrained("nvidia/segformer-b0-finetuned-ade-512-512")
-feature_extractor = SegformerFeatureExtractor.from_pretrained("nvidia/segformer-b0-finetuned-ade-512-512")
+model = SegformerForSemanticSegmentation.from_pretrained("nvidia/segformer-b3-finetuned-cityscapes-1024-1024")
+feature_extractor = SegformerFeatureExtractor.from_pretrained("nvidia/segformer-b3-finetuned-cityscapes-1024-1024")
 
 # 이미지를 처리하는 함수
 def predict(image):
     # 이미지를 모델에 맞게 변환
-    processed_image = image.resize((400, 600))
+    processed_image = image.resize((1024, 1024))
     inputs = feature_extractor(images=processed_image, return_tensors="pt")
     outputs = model(**inputs)
     logits = outputs.logits