From 9160c5168e7d04c7f905bf899bf75265f6de456e Mon Sep 17 00:00:00 2001
From: Nicolas Mowen <nickmowen213@gmail.com>
Date: Wed, 22 Oct 2025 20:48:47 -0600
Subject: [PATCH] Fix bugs

---
 frigate/util/classification.py                      |  1 +
 web/src/views/classification/ModelSelectionView.tsx | 10 +++++++++-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/frigate/util/classification.py b/frigate/util/classification.py
index 3cd9a2b70..ab17a9444 100644
--- a/frigate/util/classification.py
+++ b/frigate/util/classification.py
@@ -75,6 +75,7 @@ class ClassificationTrainingProcess(FrigateProcess):
         logger.info(f"Kicking off classification training for {self.model_name}.")
         dataset_dir = os.path.join(CLIPS_DIR, self.model_name, "dataset")
         model_dir = os.path.join(MODEL_CACHE_DIR, self.model_name)
+        os.makedirs(model_dir, exist_ok=True)
         num_classes = len(
             [
                 d
diff --git a/web/src/views/classification/ModelSelectionView.tsx b/web/src/views/classification/ModelSelectionView.tsx
index 07a1db455..372c47429 100644
--- a/web/src/views/classification/ModelSelectionView.tsx
+++ b/web/src/views/classification/ModelSelectionView.tsx
@@ -67,7 +67,15 @@ export default function ModelSelectionView({
   }
 
   if (classificationConfigs.length == 0) {
-    return <NoModelsView onCreateModel={() => setNewModel(true)} />;
+    return (
+      <>
+        <ClassificationModelWizardDialog
+          open={newModel}
+          onClose={() => setNewModel(false)}
+        />
+        <NoModelsView onCreateModel={() => setNewModel(true)} />;
+      </>
+    );
   }
 
   return (