Update handler.py

handler.py

@@ -27,12 +27,11 @@ class EndpointHandler():
         Return:
             A :obj:`dict`:. base64 encoded image
         """
-        prompt = data.pop("prompt", data)
-        negative_prompt = data.pop("negative_prompt", data)
+        inputs = data.pop("inputs", data)
         
         # run inference pipeline
         with autocast(device.type):
-            images = self.pipe(prompt, negative_prompt, guidance_scale=7.5, batch_size=8, height=512, width=768).images
+            images = self.pipe(inputs, guidance_scale=7.5).images
         # encode image as base 64
         buffered = BytesIO()
         base64_images = [];