udman99
/

briaa-2.0-aina-bg-rmv

Image Segmentation

Transformers.js

remove background

background-removal

legal liability

Model card Files Files and versions

udman99 commited on Jun 2

Commit

12f2dcd

·

verified ·

1 Parent(s): d36c36e

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -8,9 +8,9 @@ from io import BytesIO
 import base64
 class EndpointHandler:
-    def __init__(self, path=""):
         # Initialize the image segmentation pipeline
-        self.pipeline = AutoModelForImageSegmentation.from_pretrained('briaai/RMBG-2.0', trust_remote_code=True)
         torch.set_float32_matmul_precision(['high', 'highest'][0])
         self.image_size = (1024, 1024)

 import base64
 class EndpointHandler:
+    def __init__(self):
         # Initialize the image segmentation pipeline
+        self.pipeline = AutoModelForImageSegmentation.from_pretrained('.', trust_remote_code=True)
         torch.set_float32_matmul_precision(['high', 'highest'][0])
         self.image_size = (1024, 1024)