Update app.py

app.py

@@ -106,7 +106,16 @@ def sepia(input_img):
     }
     print(class_probabilities)
 
-    return fig
+    class_probabilities = {
+        labels_list[i]: probabilities[:, :, i].numpy() for i in range(len(labels_list))
+    }
+
+    # 각 클래스의 확률을 출력
+    for label, prob_map in class_probabilities.items():
+        print(f"{label} probabilities:")
+        print(prob_map)
+    
+    return fig, str(class_probabilities)
 
 demo = gr.Interface(fn=sepia,
                     inputs=gr.Image(shape=(400, 600)),