driscord_bot / app.py
fanderocha's picture
Upload app.py
5563a3d verified
raw
history blame contribute delete
780 Bytes
import gradio as gr
from rembg import remove
def remove_bg(input_image):
try:
output = remove(input_image.read())
return output
except Exception as e:
return f"Ошибка при удалении фона: {str(e)}"
iface = gr.Interface(
fn=remove_bg,
inputs=gr.Image(type="pil", label="Выберите изображение"), # Загрузка изображения через элемент управления File
outputs=gr.Image(type="pil"), # Отображение изображения с удаленным фоном
live=True
)
if __name__ == "__main__":
print("Откройте веб-браузер и перейдите по адресу, указанному в консоли.")
iface.launch()