diff --git a/components/image-generation-loading.tsx b/components/image-generation-loading.tsx
index c25f44a90..f96d1ec59 100644
--- a/components/image-generation-loading.tsx
+++ b/components/image-generation-loading.tsx
@@ -69,7 +69,7 @@ export const ImageGenerationLoading = memo(
           <div
             className={cn(
               "relative overflow-hidden rounded-[22px] border border-[#e1e4ea] bg-[#eaedf2]",
-              count === 1 ? "h-[320px] w-full max-w-[340px]" : "grid gap-3 p-3",
+              count === 1 ? "h-[420px] w-full max-w-[340px]" : "grid gap-3 p-3",
               count > 1 && "grid-cols-2"
             )}
           >
@@ -82,7 +82,7 @@ export const ImageGenerationLoading = memo(
               Array.from({ length: count }).map((_, index) => (
                 <div
                   key={index}
-                  className="relative h-[150px] overflow-hidden rounded-2xl border border-[#dfe3e9] bg-[#e7ebf0]"
+                  className="relative h-[180px] overflow-hidden rounded-2xl border border-[#dfe3e9] bg-[#e7ebf0]"
                 >
                   <div className="absolute inset-0 bg-gradient-to-r from-[#e7ebf0] via-[#dce2ea] to-[#e7ebf0]" />
                   <div