Update main.py

main.py

@@ -277,17 +277,21 @@ def dalle_3xl(
         response = requests.post(API_URL, headers=headers, json=payload, timeout=timeout)
         response.raise_for_status()
     except requests.exceptions.RequestException:
+        error_detail = [{"error": internal_error}]
+        custom_error_model = CustomErrorResponseModel(detail=error_detail)
         raise HTTPException(
             status_code=500,
-            detail=CustomErrorResponseModel(error=[{internal_error}])
+            detail=custom_error_model.dict(),
         )
 
     try:
         encoded_string = base64.b64encode(response.content).decode("utf-8")
     except Exception:
+        error_detail = [{"error": contact_support}]
+        custom_error_model = CustomErrorResponseModel(detail=error_detail)
         raise HTTPException(
             status_code=500,
-            detail=CustomErrorResponseModel(error=[{contact_support}])
+            detail=custom_error_model.dict(),
         )
 
     if encoded_string: