1

app.py

@@ -14,13 +14,12 @@ def segment_image(image):
     outputs = model(**inputs)
     logits = outputs.logits
 
-    # 결과 처리 및 이미지로 변환
+    # 결과 처리 및 NumPy 배열로 변환
     result = logits.argmax(dim=1)[0]
     result = result.cpu().detach().numpy()
-    result_image = Image.fromarray(result.astype(np.uint8), mode="P")
 
-    # 결과 이미지 반환
-    return result_image
+    # 결과 배열 반환
+    return result
 
 # Gradio 인터페이스 정의
 iface = gr.Interface(