Update app.py

app.py

@@ -103,6 +103,11 @@ def sepia(input_img):
         class_prob = np.mean(logits.numpy()[0][mask])
         class_probabilities[class_name] = class_prob
 
+    # 모든 물체의 확률 출력
+    print("Predicted class probabilities:")
+    for class_name, prob in class_probabilities.items():
+        print(f"{class_name}: {prob:.4f}")
+
     # 정확성이 가장 높은 물체 정보 얻기
     max_prob_class = max(class_probabilities, key=class_probabilities.get)
     max_prob_value = class_probabilities[max_prob_class]