Update app.py

app.py

@@ -119,13 +119,15 @@ def sepia(input_img):
     pred_img = pred_img.astype(np.uint8)
 
     fig = draw_plot(pred_img, seg)
-    return fig
+
+    # 수정된 부분: LabeledImage 위젯 사용
+    labeled_image = gr.LabeledImage(image=pred_img, label=labels_list)
+    return labeled_image
 
 demo = gr.Interface(fn=sepia,
                     inputs=gr.Image(shape=(400, 600)),
-                    outputs=['plot'],
+                    outputs='image',  # 변경된 출력 유형
                     examples=["side-1.jpg", "side-2.jpg", "side-3.jpg", "side-4.jpg", "side-5.jpg", "side-6.jpg"],
                     allow_flagging='never')
 
-
 demo.launch()