Update handler.py

handler.py

@@ -49,10 +49,10 @@ class EndpointHandler():
         with autocast(device.type):
             upscaled_image = self.pipe(prompt, image).images[0]
 
-        #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}
-        return {"image": "test"}
+        return {"image": img_str}
+        #return {"image": "test"}