fix(style): model readme img render

This commit is contained in:
jialin
2024-09-30 16:33:38 +08:00
parent 5f77a0082c
commit ea25291d75
10 changed files with 190 additions and 109 deletions
+28 -2
View File
@@ -150,7 +150,10 @@ const ModelCard: React.FC<{
target="_blank"
href={`https://huggingface.co/${modelData?.id}`}
>
<IconFont type="icon-external-link"></IconFont>
<IconFont
type="icon-external-link"
className="font-size-14"
></IconFont>
</Button>
</Tooltip>
);
@@ -165,7 +168,10 @@ const ModelCard: React.FC<{
target="_blank"
href={`https://modelscope.cn/models/${modelData?.name}`}
>
<IconFont type="icon-external-link"></IconFont>
<IconFont
type="icon-external-link"
className="font-size-14"
></IconFont>
</Button>
</Tooltip>
);
@@ -173,6 +179,16 @@ const ModelCard: React.FC<{
return null;
};
const generateModeScopeImgLink = useCallback(
(imgSrc: string) => {
if (!imgSrc) {
return '';
}
return `https://modelscope.cn/api/v1/models/${modelData?.name}/repo?Revision=${modelData?.Revision}&View=true&FilePath=${imgSrc}`;
},
[modelData]
);
useEffect(() => {
getModelCardData();
}, [props.selectedModel.name]);
@@ -240,6 +256,11 @@ const ModelCard: React.FC<{
}}
>
<MarkdownViewer
generateImgLink={
modelSource === modelSourceMap.modelscope_value
? generateModeScopeImgLink
: undefined
}
content={readmeText}
theme="light"
></MarkdownViewer>
@@ -262,6 +283,11 @@ const ModelCard: React.FC<{
<div className="card-wrapper">
<Spin spinning={loading}>
<MarkdownViewer
generateImgLink={
modelSource === modelSourceMap.modelscope_value
? generateModeScopeImgLink
: undefined
}
content={readmeText}
theme="light"
></MarkdownViewer>