Show only the last confusion matrix image after training

app.py

@@ -268,9 +268,10 @@ def main():
                     st.line_chart({tag: values})
             # Show confusion matrix images if available
             if 'confusion_matrix' in ea.Tags()['images']:
-                st.subheader("TensorBoard Confusion Matrices")
-                for img in ea.Images('confusion_matrix'):
-                    st.image(img.encoded_image_string)
+                st.subheader("TensorBoard Confusion Matrix (Last Epoch)")
+                imgs = ea.Images('confusion_matrix')
+                if imgs:
+                    st.image(imgs[-1].encoded_image_string)
         else:
             st.info("No TensorBoard logs found. Please upload logs to the 'logs' directory.")