dandygio's picture
Update app.py
a5e1c12 verified
raw
history blame contribute delete
957 Bytes
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()