Spaces:

not-lain
/

background-removal

Running on Zero

not-lain commited on Aug 28

Commit

d61a499

•

1 Parent(s): ed03ade

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

 @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