not-lain
/

BiRefNet

background-removal

mask-generation

Dichotomous Image Segmentation

Camouflaged Object Detection

Salient Object Detection

pytorch_model_hub_mixin

model_hub_mixin

Model card Files Files and versions

not-lain commited on Dec 6, 2024

Commit

a214553

·

verified ·

1 Parent(s): ef7decd

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -38,7 +38,7 @@ class EndpointHandler():
       Return:
             A :obj:`list` | `dict`: will be serialized and returned
         """
-        print('data["inputs"]',data["inputs"])
         image =  load_img(data["inputs"]).convert("RGB")
         image_size = image.size
         input_images = transform_image(image).unsqueeze(0).to("cuda")

       Return:
             A :obj:`list` | `dict`: will be serialized and returned
         """
+        print('data["inputs"] = ',data["inputs"])
         image =  load_img(data["inputs"]).convert("RGB")
         image_size = image.size
         input_images = transform_image(image).unsqueeze(0).to("cuda")