a

handler.py

@@ -46,8 +46,9 @@ class EndpointHandler():
 
         # 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)
+        guidance_scale = data.pop("guidance_scale", 7.5)
+        strength = data.pop("strength", 7.5)
+        negative_prompt = data.pop("negative_prompt", None)
         # height = data.pop("height", None)
         # width = data.pop("width", None)
         # controlnet_conditioning_scale = data.pop(
@@ -74,7 +75,8 @@ class EndpointHandler():
         out = self.pipe(
             prompt=prompt,
             image=image,
-            strength=0.75,
+            negative_prompt=negative_prompt,
+            strength=strength,
             guidance_scale=7.5,
             generator=self.generator
         )
@@ -88,5 +90,4 @@ class EndpointHandler():
         buffer = BytesIO(base64_image)
         image = Image.open(buffer).convert("RGB")
 
-
         return image