diff --git a/app/service/generate_image/service_generate_product_image.py b/app/service/generate_image/service_generate_product_image.py
index 0507953..22f7306 100644
--- a/app/service/generate_image/service_generate_product_image.py
+++ b/app/service/generate_image/service_generate_product_image.py
@@ -263,7 +263,8 @@ class GenerateProductImage:
     def get_result(self):
         try:
             prompts = [self.prompt] * self.batch_size
-            self.image = cv2.cvtColor(self.image, cv2.COLOR_BGR2RGB)
+
+            self.image = cv2.cvtColor(self.image, cv2.COLOR_RGBA2RGB)
             # self.image = cv2.resize(self.image, (1024, 1024))
             images = [self.image.astype(np.uint8)] * self.batch_size