finish

app.py

@@ -67,7 +67,6 @@ def draw_plot(pred_img, seg):
     plt.imshow(pred_img)
     plt.axis('off')
 
-    # 라벨 이름을 추가하기 위한 코드
     LABEL_NAMES = np.asarray(labels_list)
     FULL_LABEL_MAP = np.arange(len(LABEL_NAMES)).reshape(len(LABEL_NAMES), 1)
     FULL_COLOR_MAP = label_to_color_image(FULL_LABEL_MAP)
@@ -80,16 +79,6 @@ def draw_plot(pred_img, seg):
     plt.xticks([], [])
     ax.tick_params(width=0.0, labelsize=25)
 
-    # 라벨 이름 텍스트 추가
-    for label, color in enumerate(colormap):
-        mask = seg.numpy() == label
-        if np.any(mask):
-            y, x = np.where(mask)
-            y = np.mean(y).astype(int)
-            x = np.mean(x).astype(int)
-            label_name = LABEL_NAMES[label]
-            plt.text(x, y, label_name, fontsize=20, color=tuple(color))
-
     return fig
 
 
@@ -120,7 +109,7 @@ def sepia(input_img):
 
 demo = gr.Interface(fn=sepia,
                     inputs=gr.Image(shape=(1024, 1024)),
-                    outputs=['plot'],
+                    outputs=['plot', 'text'],
                     examples=["city-1.jpg", "city-2.jpg", "city-3.jpg", "city-4.jpg", "city-5.jpg"],
                     allow_flagging='never')