add test print

handler.py

@@ -70,21 +70,6 @@ class EndpointHandler():
         # Load StableDiffusionControlNetPipeline 
         #self.stable_diffusion_id = "runwayml/stable-diffusion-v1-5"
         self.stable_diffusion_id = "Lykon/dreamshaper-8"
-#        self.pipe = StableDiffusionControlNetPipeline.from_pretrained(self.stable_diffusion_id, 
-#                                                                      controlnet=self.controlnet, 
-#                                                                      torch_dtype=dtype,
-#                                                                      #safety_checker=None).to(device)
-#                                                                      #processor = AutoProcessor.from_pretrained("CompVis/stable-diffusion-safety-checker")
-#                                                                      #safety_checker = SafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
-#                                                                      safety_checker = StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
-
-#        self.pipe = StableDiffusionControlNetPipeline.from_pretrained(
-#            self.stable_diffusion_id, 
-#            controlnet=self.controlnet, 
-#            torch_dtype=dtype,
-#            safety_checker = SafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
-#        ).to(device)
-
 
         self.pipe = StableDiffusionControlNetPipeline.from_pretrained(self.stable_diffusion_id,
                                                                       controlnet=self.controlnet, 
@@ -124,6 +109,11 @@ class EndpointHandler():
         width = data.pop("width", None)
         controlnet_conditioning_scale = data.pop("controlnet_conditioning_scale", 1.0)
         
+        test_var = data.pop("test_var", "DEFAULT")
+        
+        print(f"prompt: {prompt}")
+        print(f"prompt: {test_var}")
+        
         # process image
         image = self.decode_base64_image(image)
         #control_image = CONTROLNET_MAPPING[self.control_type]["hinter"](image)