From ac3485d59eb68668da0b7b0f3a900014e95c0b8d Mon Sep 17 00:00:00 2001
From: jordan <jorraan@jorraan.au>
Date: Sat, 10 Dec 2022 22:11:44 +1100
Subject: [PATCH] Fix typos

---
 DiffuseAPI.py   | 12 ++++++------
 classweeb.ipynb |  8 ++++----
 2 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/DiffuseAPI.py b/DiffuseAPI.py
index 5164f75..51c2aac 100644
--- a/DiffuseAPI.py
+++ b/DiffuseAPI.py
@@ -43,7 +43,7 @@ class DiffuseAPI():
             return False
     
     def set_styles(self, styles):
-        if type styles = list:
+        if type(styles) == list:
             self.styles = styles
             return True
         return False
@@ -74,7 +74,7 @@ class DiffuseAPI():
         return ("portrait", self.width, self.height)
     
     def set_nsfw_filter(self, filter_state):
-        if type filter_state = bool:
+        if type(filter_state) == bool:
             self.nsfw_enabled = filter_state
             return True
         return False
@@ -85,9 +85,9 @@ class DiffuseAPI():
     async def generate_image(self, prompt, neg_prompt=""):
         payload = {
         "prompt": prompt,
-        "styles": main_settings.styles,
-        "steps": main_settings.num_steps,
-        "seed": main_settings.ai_seed,
+        "styles": self.styles,
+        "steps": self.num_steps,
+        "seed": self.ai_seed,
         "n_iter": 1,
         "height": self.height,
         "width": self.width,
@@ -105,7 +105,7 @@ class DiffuseAPI():
 
         payload.update(override_payload)
 
-        sess = aiohttp.ClientSession(main_settings.url)
+        sess = aiohttp.ClientSession(self.url)
         alive = await sess.head('/')
 
         if alive.status != 200:
diff --git a/classweeb.ipynb b/classweeb.ipynb
index 87826af..8535598 100644
--- a/classweeb.ipynb
+++ b/classweeb.ipynb
@@ -100,9 +100,9 @@
     "    async def generate_image(self, prompt, neg_prompt=\"\"):\n",
     "        payload = {\n",
     "        \"prompt\": prompt,\n",
-    "        \"styles\": main_settings.styles,\n",
-    "        \"steps\": main_settings.num_steps,\n",
-    "        \"seed\": main_settings.ai_seed,\n",
+    "        \"styles\": self.styles,\n",
+    "        \"steps\": self.num_steps,\n",
+    "        \"seed\": self.ai_seed,\n",
     "        \"n_iter\": 1,\n",
     "        \"height\": self.height,\n",
     "        \"width\": self.width,\n",
@@ -120,7 +120,7 @@
     "\n",
     "        payload.update(override_payload)\n",
     "\n",
-    "        sess = aiohttp.ClientSession(main_settings.url)\n",
+    "        sess = aiohttp.ClientSession(self.url)\n",
     "        alive = await sess.head('/')\n",
     "\n",
     "        if alive.status != 200:\n",