File size: 560 Bytes
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
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",
             examples = ["example.jpg"]  
            ).launch();