Update app.py

app.py

@@ -100,7 +100,8 @@ def sepia(input_img):
     for label in unique_labels:
         mask = (seg.numpy() == label)
         class_name = labels_list[label]
-        class_prob = np.mean(logits.numpy()[0][mask])
+        class_prob = tf.nn.softmax(logits.numpy()[0][:, :, label])  # softmax 적용
+        class_prob = np.mean(class_prob[mask])
         class_probabilities[class_name] = class_prob
 
     # Gradio Interface에 출력할 문자열 생성
@@ -123,12 +124,3 @@ demo = gr.Interface(fn=sepia,
 
 demo.launch()
 
-
-
-demo = gr.Interface(fn=sepia,
-                    inputs=gr.Image(shape=(400, 600)),
-                    outputs=['plot', 'text'],
-                    examples=["citiscapes-1.jpeg", "citiscapes-2.jpeg"],
-                    allow_flagging='never')
-
-demo.launch()