diff --git a/web/screens/Settings/HuggingFaceRepoDetailModal/ModelSegmentInfo/index.tsx b/web/screens/Settings/HuggingFaceRepoDetailModal/ModelSegmentInfo/index.tsx index 11086d237..6a6601b33 100644 --- a/web/screens/Settings/HuggingFaceRepoDetailModal/ModelSegmentInfo/index.tsx +++ b/web/screens/Settings/HuggingFaceRepoDetailModal/ModelSegmentInfo/index.tsx @@ -13,11 +13,12 @@ const ModelSegmentInfo: React.FC = () => { ) const { author, modelName, downloads, modelUrl } = useMemo(() => { - const author = - (importingHuggingFaceRepoData?.cardData['model_creator'] as string) ?? - 'N/A' - const modelName = - (importingHuggingFaceRepoData?.cardData['model_name'] as string) ?? 'N/A' + const cardData = importingHuggingFaceRepoData?.cardData + const author = (cardData?.['model_creator'] ?? 'N/A') as string + const modelName = (cardData?.['model_name'] ?? + importingHuggingFaceRepoData?.id ?? + 'N/A') as string + const modelUrl = importingHuggingFaceRepoData?.modelUrl ?? 'N/A' const downloads = importingHuggingFaceRepoData?.downloads ?? 0