From 2848c8397c5496243e571b3ad2effd988083da35 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Wed, 10 May 2023 00:14:32 +1000
Subject: [PATCH] fix(ui): fix missing images on reload issue

- Mainly an issue for commercial due to incomplete metadata handling
---
 .../src/features/gallery/components/CurrentImagePreview.tsx   | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/invokeai/frontend/web/src/features/gallery/components/CurrentImagePreview.tsx b/invokeai/frontend/web/src/features/gallery/components/CurrentImagePreview.tsx
index bbc4cd1a58..1f18fb9b9a 100644
--- a/invokeai/frontend/web/src/features/gallery/components/CurrentImagePreview.tsx
+++ b/invokeai/frontend/web/src/features/gallery/components/CurrentImagePreview.tsx
@@ -60,8 +60,8 @@ const CurrentImagePreview = () => {
         <Image
           onDragStart={handleDragStart}
           src={shouldHidePreview ? undefined : getUrl(image.url)}
-          width={image.metadata.width}
-          height={image.metadata.height}
+          width={image.metadata.width || 'auto'}
+          height={image.metadata.height || 'auto'}
           fallback={shouldHidePreview ? <CurrentImageHidden /> : undefined}
           sx={{
             objectFit: 'contain',