removed useless character from the file, removed specific endpoint url from endpoint tester

endpoint_tester.py

@@ -5,7 +5,7 @@ import base64
 from PIL import Image
 from io import BytesIO
 
-ENDPOINT_URL = "https://j0thokqylseue22z.us-east-1.aws.endpoints.huggingface.cloud" # your endpoint url
+ENDPOINT_URL = "https://XXXXXX" # your endpoint url
 HF_TOKEN = "#######################" # your huggingface token `hf_xxx`
 
 # helper image utils

handler.py

@@ -79,7 +79,6 @@ class EndpointHandler():
             reverse_norm_std = torch.tensor([0.26862954, 0.26130258, 0.27577711])[:, None, None]
             image_tensor = image_tensor * reverse_norm_std + reverse_norm_mean
             pil_img = T.ToPILImage()(image_tensor)
-            a
             image_h = pil_img.height
             image_w = pil_img.width
             image = np.array(pil_img)[:, :, [2, 1, 0]]
@@ -256,5 +255,5 @@ class EndpointHandler():
     def decode_base64_image(self, image_string):
         base64_image = base64.b64decode(image_string)
         buffer = BytesIO(base64_image)
-        image = Image.open(buffer)
+        image = Image.open(buffer).convert("RGB")
         return image