diff --git a/DiffuseAPI.py b/DiffuseAPI.py index 9c0d839..b9123cd 100644 --- a/DiffuseAPI.py +++ b/DiffuseAPI.py @@ -115,7 +115,7 @@ class DiffuseAPI(): async with aiohttp.ClientSession(self.url) as session: async with session.head('/') as alive: if alive.status != 200: - return None + return None async with session.post("/sdapi/v1/txt2img", json=payload) as image_json: image_data = await image_json.json() return image_data["images"][0] \ No newline at end of file