Spaces:

not-lain
/

background-removal

Running on Zero

not-lain commited on Aug 28

Commit

7711831

•

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