Update render.py

render.py

@@ -77,7 +77,11 @@ def generate_images(positive_prompt, image):
     ws.connect(ws_url)
     data = json.loads(json_data)
     data["49"]["inputs"]["text"] = positive_prompt
-    data["90"]["inputs"]["images"]["base64"] = [pil_to_base64(image)]
+        if image:
+        data["90"]["inputs"]["images"]["base64"] = [pil_to_base64(image)]
+    else:
+        data.pop("90", None)
+        data["68"]["enabled"] = False
     seed = random.randint(1, 1000000000)
     data["47"]["inputs"]["noise_seed"] = seed
     images = get_images(ws, data)