1

app.py

@@ -16,6 +16,7 @@ def create_color_map(num_classes):
 
 def segment_image(image):
     # 이미지 처리
+    image = image.resize(512,512)
     inputs = feature_extractor(images=image, return_tensors="pt")
     with torch.no_grad():
         outputs = model(**inputs)