background-remover

Running

not-lain commited on Aug 28, 2024

Commit

7711831

verified ·

1 Parent(s): d61a499

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,8 +23,8 @@ transform_image = transforms.Compose(
 @spaces.GPU
 def fn(image):
-    if image == None :
-        return (None,None)
     im = load_img(image, output_type="pil")
     im = im.convert("RGB")
     image_size = im.size

 @spaces.GPU
 def fn(image):
+    if not image or not image[0]:
+        return image
     im = load_img(image, output_type="pil")
     im = im.convert("RGB")
     image_size = im.size