abc

handler.py

@@ -37,21 +37,21 @@ class EndpointHandler():
         """
         prompt = data.pop("inputs", None)
         image = data.pop("image", None)
-        strength = data.pop("strength", None)
-        steps = data.pop("steps", None)
+        # strength = data.pop("strength", None)
+        # steps = data.pop("steps", None)
 
         # Check if neither prompt nor image is provided
         if prompt is None and image is None:
             return {"error": "Please provide a prompt and base64 encoded image."}
 
         # hyperparamters
-        num_inference_steps = data.pop("num_inference_steps", 30)
-        guidance_scale = data.pop("guidance_scale", 7.5)
-        negative_prompt = data.pop("negative_prompt", None)
-        height = data.pop("height", None)
-        width = data.pop("width", None)
-        controlnet_conditioning_scale = data.pop(
-            "controlnet_conditioning_scale", 1.0)
+        # num_inference_steps = data.pop("num_inference_steps", 30)
+        # guidance_scale = data.pop("guidance_scale", 7.5)
+        # negative_prompt = data.pop("negative_prompt", None)
+        # height = data.pop("height", None)
+        # width = data.pop("width", None)
+        # controlnet_conditioning_scale = data.pop(
+        #     "controlnet_conditioning_scale", 1.0)
 
         # process image
         image = self.decode_base64_image(image)