Upload handler.py

handler.py

@@ -56,7 +56,7 @@ def pil_to_base64(image: Image.Image, modelname: str, prompt: str, height: int,
     info.add_text("metadata", json.dumps(metadata))
     buffered = BytesIO()
     image.save(buffered, "PNG", pnginfo=info)
-    return base64.b64encode(buffered.getvalue())
+    return base64.b64encode(buffered.getvalue()).decode('ascii')
 
 def load_te2(repo_id: str, dtype: torch.dtype) -> Any:
     if IS_4BIT:
@@ -222,7 +222,6 @@ class EndpointHandler:
         end = time.time()
         print(f'Elapsed {end - start:.3f} sec. / prompt:"{prompt}" / size:{width}x{height} / steps:{num_inference_steps} / guidance scale:{guidance_scale} / seed:{seed}')
 
-        return image
-        #return pil_to_base64(image, self.repo_id, prompt, height, width, num_inference_steps, guidance_scale, seed)
+        return pil_to_base64(image, self.repo_id, prompt, height, width, num_inference_steps, guidance_scale, seed)