File size: 957 Bytes
69426ba
a5e1c12
 
 
69426ba
a5e1c12
 
 
69426ba
a5e1c12
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import gradio as gr
import os
from PIL import Image
import imageio

# ์ €์žฅ ๊ฒฝ๋กœ
UPLOAD_DIR = "uploads"
os.makedirs(UPLOAD_DIR, exist_ok=True)

def save_images(image1, image2):
    path1 = os.path.join(UPLOAD_DIR, "cut1.png")
    path2 = os.path.join(UPLOAD_DIR, "cut2.png")

    image1.save(path1)
    image2.save(path2)

    # ์—ฌ๊ธฐ์„œ๋Š” ์˜ˆ์‹œ๋กœ ๋‘ ์ด๋ฏธ์ง€ ๊ทธ๋Œ€๋กœ ์ด์–ด๋ถ™์ธ gif๋ฅผ ๋ฆฌํ„ด
    gif_path = os.path.join(UPLOAD_DIR, "output.gif")
    imageio.mimsave(gif_path, [image1, image2], fps=1)

    return gif_path

iface = gr.Interface(
    fn=save_images,
    inputs=[
        gr.Image(label="์ปท1 ์—…๋กœ๋“œ", type="pil"),
        gr.Image(label="์ปท2 ์—…๋กœ๋“œ", type="pil")
    ],
    outputs=gr.Video(label="๊ฒฐ๊ณผ ์• ๋‹ˆ๋ฉ”์ด์…˜"),
    title="๐ŸŒ€ ์ปท ๊ธฐ๋ฐ˜ ์• ๋‹ˆ๋ฉ”์ด์…˜ ์ƒ์„ฑ๊ธฐ (์˜ˆ์‹œ)",
    description="๋‘ ์žฅ์˜ ์ปท์„ ์—…๋กœ๋“œํ•˜๋ฉด, ์• ๋‹ˆ๋ฉ”์ด์…˜์œผ๋กœ ์ด์–ด์ฃผ๋Š” ๊ฒฐ๊ณผ๋ฅผ ๋ณด์—ฌ์ค๋‹ˆ๋‹ค."
)

iface.launch()