diff --git a/components/tabs/textImagePanel.js b/components/tabs/textImagePanel.js
index 547355d..4ede25d 100644
--- a/components/tabs/textImagePanel.js
+++ b/components/tabs/textImagePanel.js
@@ -68,7 +68,7 @@ export default function TextImagePanel({ filters }) {
     `/content/rokn/t2i?model_key=${filters?.tab_menu}`,
     postRequest,
     {
-      onSuccess: (data) => console.log(data),
+      onSuccess: (data) => setImages(data?.images),
     },
   );
 
@@ -263,7 +263,7 @@ export default function TextImagePanel({ filters }) {
       <Divider my={"20px"} borderColor={"gray.300"} />
       <ResultBox
         type="image"
-        loading={isGeneratingImage}
+        loading={isGeneratingImage || isGeneratingRoknImage}
         resultData={images}
         imgWidth={width || 500}
         imgHeight={height || 500}