File size: 537 Bytes
c862acb
c863d5e
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
23
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")
    
    width = image.width()
    height = image.height()

    image.paste(watermark, (width, height), watermark)
    return image

gr.Interface(fn=add_watermark,
             inputs="image",
             outputs="image",
             title="Add watermark" 
            ).launch();