1

app.py

@@ -18,7 +18,8 @@ def segment_image(image):
     result = logits.argmax(dim=1)[0]
     result = result.cpu().detach().numpy()
 
-    result_image = Image.fromarray(result.astype(np.uint8), mode="P")
+    result_image = Image.fromarray(result.astype(np.uint8), mode="P").convert("RGB")
+
     # 결과 배열 반환
     return result_image