stuff

DiffuseAPI.py

@@ -1,5 +1,4 @@
 import aiohttp
-from base64 import b64decode
 from sys import exit
 
 class DiffuseAPI():
@@ -83,18 +82,6 @@ class DiffuseAPI():
         return self.nsfw_enabled
 
     async def generate_image(self, prompt, neg_prompt=""):
-        payload = {
-        "prompt": prompt,
-        "styles": self.styles,
-        "steps": self.num_steps,
-        "seed": self.seed,
-        "n_iter": 1,
-        "height": self.height,
-        "width": self.width,
-        "negative_prompts": neg_prompt,
-        "cfg_scale": self.cfg_scale
-        }
-
         settings = {
         "filter_nsfw": not self.nsfw_enabled
         }
@@ -103,7 +90,9 @@ class DiffuseAPI():
             "override_settings": settings
         }
 
-        payload.update(override_payload)
+        payload = {"prompt": prompt, "styles": self.styles, "steps": self.num_steps, "seed": self.seed,
+                    "n_iter": 1, "height": self.height, "width": self.width, "negative_prompts": neg_prompt,
+                    "cfg_scale": self.cfg_scale} | override_payload
 
         sess = aiohttp.ClientSession(self.url)
         alive = await sess.head('/')
@@ -118,4 +107,4 @@ class DiffuseAPI():
                     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]
+                return image_data["images"][0]