whlzy
/

remove_bg_api

Image Segmentation

Transformers.js

remove background

background-removal

legal liability

Model card Files Files and versions Community

whlzy commited on 6 days ago

Commit

e928816

·

verified ·

1 Parent(s): 1104805

Upload handler.py

Files changed (1) hide show

handler.py +4 -2

handler.py CHANGED Viewed

@@ -43,8 +43,10 @@ class EndpointHandler():
             aws_secret_access_key=SECRET_ACCESS_KEY,
             region_name='auto'
         )
-        image.seek(0)
-        s3.upload_fileobj(image, BUCKET_NAME, f"{BUCKET_PREFIX_PATH}/{bucket_postfix_path}")
         return image_url
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]:

             aws_secret_access_key=SECRET_ACCESS_KEY,
             region_name='auto'
         )
+        output_buffer = BytesIO()
+        img_pil.save(output_buffer, format='WEBP', quality=85, method=4)
+        output_buffer.seek(0)
+        s3.upload_fileobj(output_buffer, BUCKET_NAME, f"{BUCKET_PREFIX_PATH}/{bucket_postfix_path}")
         return image_url
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]: