Compare commits

..

3 Commits

Author SHA1 Message Date
Nicolas Mowen
0f9ce04b1a Check length 2025-10-22 07:29:04 -06:00
Nicolas Mowen
070faa0d49 Don't filter all configs 2025-10-22 07:15:33 -06:00
Nicolas Mowen
60462b9b45 Set unknown 2025-10-22 07:05:30 -06:00
2 changed files with 9 additions and 7 deletions

View File

@ -219,10 +219,10 @@ export function GroupedClassificationCard({
const bestTyped: ClassificationItemData = best; const bestTyped: ClassificationItemData = best;
return { return {
...bestTyped, ...bestTyped,
name: event ? (event.sub_label ?? "") : bestTyped.name, name: event ? (event.sub_label ?? t("details.unknown")) : bestTyped.name,
score: event?.data?.sub_label_score || bestTyped.score, score: event?.data?.sub_label_score || bestTyped.score,
}; };
}, [group, event]); }, [group, event, t]);
const bestScoreStatus = useMemo(() => { const bestScoreStatus = useMemo(() => {
if (!bestItem?.score || !threshold) { if (!bestItem?.score || !threshold) {

View File

@ -39,9 +39,11 @@ export default function ModelSelectionView({
return []; 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) { if (pageToggle == "objects" && model.object_config != undefined) {
return true; return true;
} }
@ -52,7 +54,7 @@ export default function ModelSelectionView({
return false; return false;
}); });
}, [config, pageToggle]); }, [classificationConfigs, pageToggle]);
// new model wizard // new model wizard
@ -115,7 +117,7 @@ export default function ModelSelectionView({
</div> </div>
</div> </div>
<div className="flex size-full gap-2 p-2"> <div className="flex size-full gap-2 p-2">
{classificationConfigs.map((config) => ( {selectedClassificationConfigs.map((config) => (
<ModelCard <ModelCard
key={config.name} key={config.name}
config={config} config={config}
@ -137,7 +139,7 @@ function ModelCard({ config, onClick }: ModelCardProps) {
}>(`classification/${config.name}/dataset`, { revalidateOnFocus: false }); }>(`classification/${config.name}/dataset`, { revalidateOnFocus: false });
const coverImage = useMemo(() => { const coverImage = useMemo(() => {
if (!dataset) { if (!dataset?.length) {
return undefined; return undefined;
} }