Improve model train button

This commit is contained in:
Nicolas Mowen 2025-07-16 15:48:09 -06:00
parent 99885b4bdc
commit af5f3e5f84
2 changed files with 9 additions and 3 deletions

View File

@ -3,7 +3,8 @@
"deleteClassificationAttempts": "Delete Classification Images", "deleteClassificationAttempts": "Delete Classification Images",
"renameCategory": "Rename Class", "renameCategory": "Rename Class",
"deleteCategory": "Delete Class", "deleteCategory": "Delete Class",
"deleteImages": "Delete Images" "deleteImages": "Delete Images",
"trainModel": "Train Model"
}, },
"toast": { "toast": {
"success": { "success": {

View File

@ -50,6 +50,7 @@ import { ModelState } from "@/types/ws";
import ActivityIndicator from "@/components/indicators/activity-indicator"; import ActivityIndicator from "@/components/indicators/activity-indicator";
import { useNavigate } from "react-router-dom"; import { useNavigate } from "react-router-dom";
import { IoMdArrowRoundBack } from "react-icons/io"; import { IoMdArrowRoundBack } from "react-icons/io";
import { MdAutoFixHigh } from "react-icons/md";
type ModelTrainingViewProps = { type ModelTrainingViewProps = {
model: CustomClassificationModelConfig; model: CustomClassificationModelConfig;
@ -345,8 +346,12 @@ export default function ModelTrainingView({ model }: ModelTrainingViewProps) {
onClick={trainModel} onClick={trainModel}
disabled={modelState != "complete"} disabled={modelState != "complete"}
> >
Train Model {modelState == "training" ? (
{modelState == "training" && <ActivityIndicator size={20} />} <ActivityIndicator size={20} />
) : (
<MdAutoFixHigh />
)}
{t("button.trainModel")}
</Button> </Button>
)} )}
</div> </div>