diff --git a/backend/dist/nsfwapp-linux-amd64 b/backend/dist/nsfwapp-linux-amd64
index 5a337ed..3d840df 100644
Binary files a/backend/dist/nsfwapp-linux-amd64 and b/backend/dist/nsfwapp-linux-amd64 differ
diff --git a/backend/dist/nsfwapp.exe b/backend/dist/nsfwapp.exe
index b346209..4d70ade 100644
Binary files a/backend/dist/nsfwapp.exe and b/backend/dist/nsfwapp.exe differ
diff --git a/backend/dist/nsfwapp_amd64.deb b/backend/dist/nsfwapp_amd64.deb
index 19f8cd7..1ffc11f 100644
Binary files a/backend/dist/nsfwapp_amd64.deb and b/backend/dist/nsfwapp_amd64.deb differ
diff --git a/frontend/src/components/ui/TrainingTab.tsx b/frontend/src/components/ui/TrainingTab.tsx
index eb2be13..4e946b3 100644
--- a/frontend/src/components/ui/TrainingTab.tsx
+++ b/frontend/src/components/ui/TrainingTab.tsx
@@ -6988,7 +6988,7 @@ export default function TrainingTab(props: {
     setLoadingPreviewFailed(false)
   }, [loading, frameBusy, loadingPreviewFallbackUrl, loadingPreviewUrl])
 
-  const showImageBoxes = !frameBusy && !trainingRunning
+  const showImageBoxes = !frameBusy && !trainingRunning && !saving
 
   const shownTrainingDurationMs = useMemo(() => {
     const job = trainingStatus?.training