Merge pull request #15 from nathanfdunn/main

Handle case when request is rejected by Dalle

Author: ezzcodeezzlife

src/dalle2/dalle2.py

@@ -82,6 +82,9 @@ def get_task_response(self, body):
             if data["status"] == "failed":
                 print(f"Task failed: {data['status_information']}")
                 return None
+            if data["status"] == "rejected":
+                print(f"Task rejected: {data['status_information']}")
+                return None
             if data["status"] == "succeeded":
                 print("🙌 Task completed!")
                 return data["generations"]["data"]