Update handler.py

handler.py

@@ -28,19 +28,27 @@ class EndpointHandler():
             A :obj:`dict`:. base64 encoded image
         """
         inputs = data.pop("inputs", data)
+        print("Printing inputs")
         print(inputs)
+        print("")
+
+        print("Printing image")
+        print(inputs['image'])
+        print("")
 
         # decode base64 image to PIL
         image = Image.open(BytesIO(base64.b64decode(inputs['image'])))
+        print("Printing loaded image into library")
         print(image)
+        print("")
         
         # run inference pipeline
-        upscaled_image = self.pipe(prompt="", image = image).images[0]
+        #upscaled_image = self.pipe(prompt="", image = image).images[0]
 
         # encode image as base 64
-        buffered = BytesIO()
-        upscaled_image.save(buffered, format="JPEG")
-        img_str = base64.b64encode(buffered.getvalue())
+        #buffered = BytesIO()
+        #upscaled_image.save(buffered, format="JPEG")
+        #img_str = base64.b64encode(buffered.getvalue())
 
         # postprocess the prediction
-        return {"image": img_str.decode()}
+        return {"image": "img_str.decode()"}