Update handler.py

handler.py

@@ -33,8 +33,11 @@ class EndpointHandler:
         self.pipe_refiner.enable_model_cpu_offload()
 
     def __call__(self, data: Dict[str, Any]) -> Dict[str, str]:
+        print(data)
+        inputs = data.pop("inputs", data)
+
         print(inputs)
-        prompt, prompt2, negative_prompt, negative_prompt2, height = inputs['prompt'], inputs['prompt2'], inputs['negative_prompt'], inputs['negative_prompt2'], inputs['height']
+        prompt, prompt2, negative_prompt, negative_prompt2 = inputs['prompt'], inputs['prompt2'], inputs['negative_prompt'], inputs['negative_prompt2']
 
         image_base_latent = self.pipe(
                             prompt=prompt,