File size: 529 Bytes
c862acb c863d5e c862acb edb0e97 c862acb 42dcc6e c862acb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import gradio as gr
import numpy as np
from PIL import Image
def add_watermark(image):
image = Image.fromarray(np.uint8(image))
watermark = Image.open("watermark.png")
image = image.convert("RGBA")
watermark = watermark.convert("RGBA")
watermark = watermark.resize((image.width, image.height))
image.paste(watermark,(0,0),watermark)
return image
gr.Interface(fn=add_watermark,
inputs="image",
outputs="image",
title="Add watermark"
).launch(); |