From 070faa0d49670cda84fa95a3b4cf79eea9dfcc4c Mon Sep 17 00:00:00 2001
From: Nicolas Mowen <nickmowen213@gmail.com>
Date: Wed, 22 Oct 2025 07:15:33 -0600
Subject: [PATCH] Don't filter all configs

---
 web/src/views/classification/ModelSelectionView.tsx | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/web/src/views/classification/ModelSelectionView.tsx b/web/src/views/classification/ModelSelectionView.tsx
index 53f9c0af0..42f7c5773 100644
--- a/web/src/views/classification/ModelSelectionView.tsx
+++ b/web/src/views/classification/ModelSelectionView.tsx
@@ -39,9 +39,11 @@ export default function ModelSelectionView({
       return [];
     }
 
-    const allModels = Object.values(config.classification.custom);
+    return Object.values(config.classification.custom);
+  }, [config]);
 
-    return allModels.filter((model) => {
+  const selectedClassificationConfigs = useMemo(() => {
+    return classificationConfigs.filter((model) => {
       if (pageToggle == "objects" && model.object_config != undefined) {
         return true;
       }
@@ -52,7 +54,7 @@ export default function ModelSelectionView({
 
       return false;
     });
-  }, [config, pageToggle]);
+  }, [classificationConfigs, pageToggle]);
 
   // new model wizard
 
@@ -115,7 +117,7 @@ export default function ModelSelectionView({
         </div>
       </div>
       <div className="flex size-full gap-2 p-2">
-        {classificationConfigs.map((config) => (
+        {selectedClassificationConfigs.map((config) => (
           <ModelCard
             key={config.name}
             config={config}