Spaces:

not-lain
/

background-removal

Running on Zero

not-lain commited on Aug 29

Commit

2b841c3

•

1 Parent(s): 1a7230f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,7 +23,7 @@ transform_image = transforms.Compose(
 @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")

 @spaces.GPU
 def fn(image):
+    if not image:
         return image
     im = load_img(image, output_type="pil")
     im = im.convert("RGB")