Spaces:
Running
on
Zero
Running
on
Zero
fix resize
Browse files- demo/rm_bg.py +1 -1
demo/rm_bg.py
CHANGED
|
@@ -5,7 +5,7 @@ import rembg
|
|
| 5 |
|
| 6 |
def rm_bg(img, use_sam=False):
|
| 7 |
img = Image.fromarray(img)
|
| 8 |
-
img.resize((512, 512))
|
| 9 |
output = rembg.remove(img)
|
| 10 |
mask = np.array(output)[:, :, 3]
|
| 11 |
|
|
|
|
| 5 |
|
| 6 |
def rm_bg(img, use_sam=False):
|
| 7 |
img = Image.fromarray(img)
|
| 8 |
+
img = img.resize((512, 512))
|
| 9 |
output = rembg.remove(img)
|
| 10 |
mask = np.array(output)[:, :, 3]
|
| 11 |
|