diff --git a/core/src/types/index.ts b/core/src/types/index.ts index 7a7faad6d5..dcbc20ca32 100644 --- a/core/src/types/index.ts +++ b/core/src/types/index.ts @@ -240,6 +240,7 @@ export type ModelMetadata = { author: string; tags: string[]; size: number; + cover?: string; }; /** diff --git a/web/screens/ExploreModels/ExploreModelItemHeader/index.tsx b/web/screens/ExploreModels/ExploreModelItemHeader/index.tsx index 3621c90ec6..79428b0423 100644 --- a/web/screens/ExploreModels/ExploreModelItemHeader/index.tsx +++ b/web/screens/ExploreModels/ExploreModelItemHeader/index.tsx @@ -98,23 +98,30 @@ const ExploreModelItemHeader: React.FC = ({ model, onClick, open }) => { return (
-
- {model.name} -
-
- - {toGigabytes(model.metadata.size)} - - {downloadButton} - + {model.metadata.cover && ( +
+ {`Cover +
+ )} +
+
+ {model.name} +
+
+ + {toGigabytes(model.metadata.size)} + + {downloadButton} + +
)