update handler

handler.py

@@ -156,10 +156,13 @@ class EndpointHandler():
             "seed": -1,
         }
         if data["inputs"]:
-             if "prompt" in data["inputs"].keys():
-                prompt = data["inputs"]["prompt"]
-                print("get prompt from request: ", prompt)
-                args["prompt"] = prompt
+            for field, v in args:
+                if data["inputs"][field]:
+                    args[field] = data["inputs"][field]
+             # if "prompt" in data["inputs"].keys():
+             #    prompt = data["inputs"]["prompt"]
+             #    print("get prompt from request: ", prompt)
+             #    args["prompt"] = prompt
         p = StableDiffusionProcessingTxt2Img(sd_model=self.shared.sd_model, **args)
         processed = process_images(p)
         single_image_b64 = encode_pil_to_base64(processed.images[0]).decode('utf-8')