app.py

from PIL import Image
import torch
import numpy as np
import gradio as gr






# AnimeGAN-V2 모델 로드
generator = torch.hub.load(
    "AK391/animegan2-pytorch:main", "generator", pretrained=True, progress=False
)
face2paint = torch.hub.load(
    "AK391/animegan2-pytorch:main", "face2paint", size=512, side_by_side=False
)


def inference(img):
    """AnimeGAN-V2를 사용하여 이미지를 애니메이션 스타일로 변환하는 함수"""
    img_pil = Image.fromarray(img.astype('uint8'), 'RGB')  # NumPy 배열을 PIL 이미지로 변환
    output_image_pil = face2paint(generator, img_pil)  # AnimeGAN-V2 변환 수행







    return np.array(output_image_pil)  # 변환된 이미지를 NumPy 배열로 반환












# Gradio UI 생성
gr.Interface(
    fn=inference,
    inputs=gr.Image(type="numpy"),
    outputs=gr.Image(type="numpy"),
    title="AnimeGAN-V2 Image Transformation",
    description="Upload an image to transform it into an anime-style image using AnimeGAN-V2.",
    live=True  # 실시간 변환 지원
).launch()