Spaces:

dhanushreddy29
/

BGRemovalTest

Sleeping

dhanushreddy29 commited on Oct 15, 2023

Commit

afbc44e

1 Parent(s): 8a7a91a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,6 +10,7 @@ from PIL import Image
 def removeBackground(image):
     if not os.path.exists("saved_models"):
         os.mkdir("saved_models")
         os.mkdir("git")

 def removeBackground(image):
+    image = Image.fromarray(image).convert("RGB")
     if not os.path.exists("saved_models"):
         os.mkdir("saved_models")
         os.mkdir("git")