diff --git a/web/public/locales/en/views/classificationModel.json b/web/public/locales/en/views/classificationModel.json
index f5ed7109f..65b0ffa1e 100644
--- a/web/public/locales/en/views/classificationModel.json
+++ b/web/public/locales/en/views/classificationModel.json
@@ -1,25 +1,25 @@
 {
   "button": {
     "deleteClassificationAttempts": "Delete Classification Images",
-    "renameCategory": "Rename Category",
-    "deleteCategory": "Rename Category",
+    "renameCategory": "Rename Class",
+    "deleteCategory": "Delete Class",
     "deleteImages": "Delete Images"
   },
   "toast": {
     "success": {
-      "deletedCategory": "Deleted Category",
+      "deletedCategory": "Deleted Class",
       "deletedImage": "Deleted Images",
       "categorizedImage": "Successfully Categorized Image"
     },
     "error": {
       "deleteImageFailed": "Failed to delete: {{errorMessage}}",
-      "deleteCategoryFailed": "Failed to delete category: {{errorMessage}}",
+      "deleteCategoryFailed": "Failed to delete class: {{errorMessage}}",
       "categorizeFailed": "Failed to categorize image: {{errorMessage}}"
     }
   },
   "deleteCategory": {
-    "title": "Delete Category",
-    "desc": "Are you sure you want to delete the category {{name}}? This will permanently delete all associated images and require re-training the model."
+    "title": "Delete Class",
+    "desc": "Are you sure you want to delete the class {{name}}? This will permanently delete all associated images and require re-training the model."
   },
   "deleteDatasetImages": {
     "title": "Delete Dataset Images",
@@ -40,9 +40,9 @@
     "title": "Train",
     "aria": "Select Train"
   },
-  "categories": "Categories",
+  "categories": "Classes",
   "createCategory": {
-    "new": "Create New Category"
+    "new": "Create New Class"
   },
   "categorizeImageAs": "Categorize Image As:",
   "categorizeImage": "Categorize Image"
diff --git a/web/src/components/overlay/ClassificationSelectionDialog.tsx b/web/src/components/overlay/ClassificationSelectionDialog.tsx
index 6ef07408c..7cb8ca156 100644
--- a/web/src/components/overlay/ClassificationSelectionDialog.tsx
+++ b/web/src/components/overlay/ClassificationSelectionDialog.tsx
@@ -32,7 +32,7 @@ import { toast } from "sonner";
 
 type ClassificationSelectionDialogProps = {
   className?: string;
-  categories: string[];
+  classes: string[];
   modelName: string;
   image: string;
   onRefresh: () => void;
@@ -40,7 +40,7 @@ type ClassificationSelectionDialogProps = {
 };
 export default function ClassificationSelectionDialog({
   className,
-  categories,
+  classes,
   modelName,
   image,
   onRefresh,
@@ -135,7 +135,7 @@ export default function ClassificationSelectionDialog({
                 <LuPlus />
                 {t("createCategory.new")}
               </SelectorItem>
-              {categories.sort().map((category) => (
+              {classes.sort().map((category) => (
                 <SelectorItem
                   key={category}
                   className="flex cursor-pointer gap-2 smart-capitalize"
diff --git a/web/src/views/classification/ModelTrainingView.tsx b/web/src/views/classification/ModelTrainingView.tsx
index f0b837929..fe4e5cc10 100644
--- a/web/src/views/classification/ModelTrainingView.tsx
+++ b/web/src/views/classification/ModelTrainingView.tsx
@@ -280,7 +280,7 @@ export default function ModelTrainingView({ model }: ModelTrainingViewProps) {
       {pageToggle == "train" ? (
         <TrainGrid
           model={model}
-          categories={Object.keys(dataset || {})}
+          classes={Object.keys(dataset || {})}
           trainImages={trainImages || []}
           selectedImages={selectedImages}
           onRefresh={refreshTrain}
@@ -561,7 +561,7 @@ function DatasetGrid({
 
 type TrainGridProps = {
   model: CustomClassificationModelConfig;
-  categories: string[];
+  classes: string[];
   trainImages: string[];
   selectedImages: string[];
   onClickImages: (images: string[], ctrl: boolean) => void;
@@ -570,7 +570,7 @@ type TrainGridProps = {
 };
 function TrainGrid({
   model,
-  categories,
+  classes,
   trainImages,
   selectedImages,
   onClickImages,
@@ -628,7 +628,7 @@ function TrainGrid({
               </div>
               <div className="flex flex-row items-start justify-end gap-5 md:gap-4">
                 <ClassificationSelectionDialog
-                  categories={categories}
+                  classes={classes}
                   modelName={model.name}
                   image={data.raw}
                   onRefresh={onRefresh}