Update handler.py

handler.py

@@ -29,14 +29,16 @@ class EndpointHandler():
         replacement=texts[1]
         if replacement=="":
             img=self.model.inference_remove(raw_image,target)
-            img_data=img.tobytes()
-            img_base64=base64.b64encode(img_data).decode('utf-8')
-            return {'image':img_base64}
+            img.save("./1.png")
+            with open('./1.png','rb') as img_file:
+                encoded_string = base64.b64encode(img_file.read()).decode('utf-8')
+            return {'image':encoded_string}
         else:
             img=self.model.inference_replace_sam(raw_image,target,replacement)
-            img_data=img.tobytes()
-            img_base64=base64.b64encode(img_data).decode('utf-8')
-            return {'image':img_base64}
+            img.save("./1.png")
+            with open('./1.png','rb') as img_file:
+                encoded_string = base64.b64encode(img_file.read()).decode('utf-8')
+            return {'image':encoded_string}
             
     
 if __name__=="__main__":