diff --git a/web/screens/HubScreen2/components/HubScreenFilter/index.tsx b/web/screens/HubScreen2/components/HubScreenFilter/index.tsx index 38b673c570..0148bb01e0 100644 --- a/web/screens/HubScreen2/components/HubScreenFilter/index.tsx +++ b/web/screens/HubScreen2/components/HubScreenFilter/index.tsx @@ -1,3 +1,7 @@ +import { Fragment } from 'react' + +import Image from 'next/image' + import BlankState from '@/containers/BlankState' import useModelHub from '@/hooks/useModelHub' @@ -43,6 +47,9 @@ const HubScreenFilter: React.FC = ({ queryText }) => { filteredHuggingFaceModels.length === 0 && filteredRemoteModels.length === 0 + const isOnDevice = + filteredBuiltInModels.length > 0 || filteredHuggingFaceModels.length > 0 + return (
{isResultEmpty ? ( @@ -50,16 +57,44 @@ const HubScreenFilter: React.FC = ({ queryText }) => {
) : ( -
- {filteredBuiltInModels.map((hfModelEntry) => ( - - ))} - {filteredHuggingFaceModels.map((hfModelEntry) => ( - - ))} - {filteredRemoteModels.map((hfModelEntry) => ( - - ))} +
+ {isOnDevice && ( + +
+ Built-In Models +

On-Device Models

+
+
+ {filteredBuiltInModels.map((hfModelEntry) => ( + + ))} + {filteredHuggingFaceModels.map((hfModelEntry) => ( + + ))} +
+
+ )} + + {filteredRemoteModels.length > 0 && ( + +
+

Cloud Models

+
+
+ {filteredRemoteModels.map((hfModelEntry) => ( + + ))} +
+
+ )}
)}
diff --git a/web/screens/HubScreen2/components/RemoteModelGroup.tsx b/web/screens/HubScreen2/components/RemoteModelGroup.tsx index 2f8b8d9b71..ae470f09d7 100644 --- a/web/screens/HubScreen2/components/RemoteModelGroup.tsx +++ b/web/screens/HubScreen2/components/RemoteModelGroup.tsx @@ -62,7 +62,13 @@ const RemoteModelGroup: React.FC = ({ data, engine, onSeeAllClick }) => {
{engineLogo && ( - Engine logo + Engine logo )}

{refinedTitle}