Update app.py

app.py

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