Update app.py

app.py

@@ -1,23 +1,15 @@
 import gradio as gr
-import cv2
-import numpy as np
+from rembg import remove
 
 # Функция для удаления водяных знаков с изображения
 def remove_watermark(input_image):
-    # Загрузка изображения
-    image = cv2.imdecode(np.frombuffer(input_image.read(), np.uint8), -1)
-    
-    # Здесь вы можете использовать библиотеки и алгоритмы для обнаружения и удаления водяных знаков.
-    # Например, можно использовать OpenCV для обработки изображения.
-    
-    # В этом примере просто уберем красные пиксели (подразумевая, что водяные знаки красные)
-    image[np.all(image == [255, 0, 0], axis=-1)] = [255, 255, 255]
+    # Загрузка изображения и удаление фона
+    with open("input.png", "wb") as f:
+        f.write(input_image.read())
 
-    # Преобразование изображения обратно в байты
-    _, encoded_image = cv2.imencode(".jpg", image)
-    result_image = encoded_image.tobytes()
+    output = remove("input.png")
 
-    return result_image
+    return output.tobytes()
 
 # Создание Gradio приложения
 iface = gr.Interface(
@@ -25,7 +17,6 @@ iface = gr.Interface(
     inputs=gr.Image(),
     outputs=gr.Image(),
     live=True,
-    capture_session=True
 )
 
 iface.launch()