1

app.py

@@ -18,8 +18,9 @@ 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")
     # 결과 배열 반환
-    return result
+    return result_image
 
 # Gradio 인터페이스 정의
 iface = gr.Interface(