diff --git a/app/api/api_generate_image.py b/app/api/api_generate_image.py
index c3332b6..5f7ff4b 100644
--- a/app/api/api_generate_image.py
+++ b/app/api/api_generate_image.py
@@ -63,10 +63,14 @@ async def generate_image_flux2_klein(request_item: GenerateImageFlux2KleinModel)
                 f"http://{settings.FLUX2_GEN_IMG_MODEL_URL}/predict",
                 json=request_item.model_dump(),
             )
-        result = resp.json()
-
-        logger.info(f"generate_image_flux2_gen_img response: {json.dumps(result, indent=4)}")
-        return ResponseModel(data=result)
+        if resp.status_code == 200:
+            result = resp.json()
+            logger.info(f"flux2_gen_img response: {json.dumps(result, indent=4)}")
+            return ResponseModel(data=result)
+        else:
+            error = resp.json()
+            logger.info(f"flux2_gen_img response: {json.dumps(error, indent=4)}")
+            return ResponseModel(data=error, msg="ERROR!", code=500)
 
     except Exception as e:
         logger.warning(f"generate_image_flux2_gen_img Run Exception @@@@@@:{e}")