From 6ba540d1f70bb42f4137544fc62ff790c42487b8 Mon Sep 17 00:00:00 2001 From: NamH Date: Tue, 21 May 2024 13:53:46 +0700 Subject: [PATCH] fix: fix crash when model name not ready (#2931) Signed-off-by: James Co-authored-by: James --- .../ModelSegmentInfo/index.tsx | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) 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