Нашел скрипт с удалением фона картинки почему то не работает. Как исправить?

Question

[User_id_cleo]

Я нашел скрипт для удаления фона с картинки, но по какой-то причине он не работает.
Возвращает ошибку:

Traceback (most recent call last):
  File "C:\Users\Александр\Desktop\CLEO_BOT\command_menu.py", line 22, in <module>
    imgs = Image.open(io.BytesIO(result)).convert("RGBA")
  File "C:\Users\Александр\AppData\Local\Programs\Python\Python39\lib\site-packages\PIL\Image.py", line 3147, in open
    raise UnidentifiedImageError(
PIL.UnidentifiedImageError: cannot identify image file <_io.BytesIO object at 0x00000144B975EA90>

>>> s
вот сам скрипт:

SOURCE_DIR = 'C:/Users/Александр/Desktop/CLEO_BOT/'
person_one = Image.open(SOURCE_DIR + "person_one.jpg")
print(person_one.size)
person_one.show()


input_path = 'person_one.jpg'
output_path = 'fon_pillow_paste_out.jpg.jpg'

f = np.fromfile(input_path)
result = remove(f)
imgs = Image.open(io.BytesIO(result)).convert("RGBA")
imgs.save(output_path)

Пожалуйста, скажите мне, где ошибка и как её исправить ?

Comments

[Vindicar]

Что в result? Конкретно - bytes, None, что-то ещё?

[Алан Гибизов]

Что в функции remove()? Что она возвращает? Если в ней нет return, то она возвращает None.

[mayton2019]

Автор покажи requirements.txt. Покажи импорты в исходнике.

[User_id_cleo]

mayton2019,

from rembg.bg import remove
import numpy as np
import io
from PIL import Image

input_path = 'in.png'
output_path = 'out.png'

f = np.fromfile(input_path)
result = remove(f)
img = Image.open(io.BytesIO(result)).convert("RGBA")
img.save(output_path)