diff --git a/src/apps/models-modal/ModelsSourceSelector.tsx b/src/apps/models-modal/ModelsSourceSelector.tsx
index 3f6160cc0..ed8ff5296 100644
--- a/src/apps/models-modal/ModelsSourceSelector.tsx
+++ b/src/apps/models-modal/ModelsSourceSelector.tsx
@@ -1,7 +1,7 @@
import * as React from 'react';
import { shallow } from 'zustand/shallow';
-import { Badge, Box, Button, IconButton, ListItemDecorator, MenuItem, Option, Select, Typography } from '@mui/joy';
+import { Avatar, Badge, Box, Button, IconButton, ListItemDecorator, MenuItem, Option, Select, Typography } from '@mui/joy';
import AddIcon from '@mui/icons-material/Add';
import CloudDoneOutlinedIcon from '@mui/icons-material/CloudDoneOutlined';
import CloudOutlinedIcon from '@mui/icons-material/CloudOutlined';
@@ -25,10 +25,16 @@ function locationIcon(vendor?: IModelVendor | null) {
}
function vendorIcon(vendor: IModelVendor | null, greenMark: boolean) {
- const Icon = !vendor ? null : vendor.Icon;
- return (greenMark && Icon)
- ?
- : Icon ? : null;
+ let icon: React.JSX.Element | null = null;
+ if (vendor?.Icon) {
+ if (typeof vendor.Icon === 'string')
+ icon = {vendor.Icon};
+ else
+ icon = ;
+ }
+ return (greenMark && icon)
+ ? {icon}
+ : icon;
}