From f893440ab712cf1fe6673a3482e97a1a38570a43 Mon Sep 17 00:00:00 2001
From: Nicolas Mowen <nickmowen213@gmail.com>
Date: Tue, 7 Oct 2025 09:24:30 -0600
Subject: [PATCH] Cleanup

---
 web/src/views/classification/ModelSelectionView.tsx | 6 +++++-
 web/src/views/classification/ModelTrainingView.tsx  | 1 +
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/web/src/views/classification/ModelSelectionView.tsx b/web/src/views/classification/ModelSelectionView.tsx
index 2dd7e0375..b91ea5648 100644
--- a/web/src/views/classification/ModelSelectionView.tsx
+++ b/web/src/views/classification/ModelSelectionView.tsx
@@ -38,7 +38,11 @@ export default function ModelSelectionView({
   return (
     <div className="flex size-full gap-2 p-2">
       {classificationConfigs.map((config) => (
-        <ModelCard config={config} onClick={() => onClick(config)} />
+        <ModelCard
+          key={config.name}
+          config={config}
+          onClick={() => onClick(config)}
+        />
       ))}
     </div>
   );
diff --git a/web/src/views/classification/ModelTrainingView.tsx b/web/src/views/classification/ModelTrainingView.tsx
index 391bc90d7..d3fd5b40e 100644
--- a/web/src/views/classification/ModelTrainingView.tsx
+++ b/web/src/views/classification/ModelTrainingView.tsx
@@ -801,6 +801,7 @@ function StateTrainGrid({
     >
       {trainData?.map((data) => (
         <ClassificationCard
+          key={data.filename}
           className="w-60 gap-2 rounded-lg bg-card p-2"
           imgClassName="size-auto"
           data={data}