Update handler.py

handler.py

@@ -27,29 +27,15 @@ class EndpointHandler():
         self.pipe = self.pipe.to(device)
 
 
-    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
+    def __call__(self, *args, **kwargs) -> List[Dict[str, Any]]:
         """
         Args:
-            images (:obj:`string`)
+            image (:obj:`string`)
         Return:
             A :obj:`dict`:. base64 encoded image
         """
-        inputs = data.pop("inputs", data)
-        logger.info(f"Printing inputs {inputs}")
-        logger.info(f"Printing image {inputs['image']}")
-
-        # decode base64 image to PIL
-        decoded_image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
-        logger.info(f"Printing loaded image into library {decoded_image}")
-        
-        # run inference pipeline
-        with autocast(device.type):
-            upscaled_image = self.pipe(prompt="", image = decoded_image).images[0]
-
-        # encode image as base 64
-        buffered = BytesIO()
-        upscaled_image.save(buffered, format="JPEG")
-        img_str = base64.b64encode(buffered.getvalue())
+        print(args)
+        print(kwargs)
 
         # postprocess the prediction
-        return {"image": img_str}
+        return {"image": "nothing"}