File size: 1,495 Bytes
376eaac
 
17aee8d
376eaac
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9f94fce
376eaac
 
 
 
a00870f
 
376eaac
 
 
 
 
 
 
 
 
b04bcb4
 
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
36
37
38
39
from PIL import Image
import torch
import torchvision
import gradio

Instructuction = "Select a Unique Portrait Image of yourself"
title="I am something of a Painter, Anime-Edition (AnimeGAN-V2)"
description = "Drag/Drop or Upload a cute portrait Image of yourself or anyone you find interesting 😉, then observe how this Generative model\

               is able to Generate a cute Anime-Cartoon version of your Image."
article = """

            - Select an image from the examples provided as demo image,

            - Click submit to Generate Image,

            - Tips: Quality Images with Great brightness/without pre-existing filters work better (Image Noise).

            - Privacy: No user data is collected or saved,

            - Credits to akhaliq/AnimeGANv2 for original AnimeGanV2"""

model = torch.hub.load(
    "AK391/animegan2-pytorch:main",
    "generator",
    pretrained=True,
    #device="cuda",
    progress=False
)
face2paint = torch.hub.load(
    'AK391/animegan2-pytorch:main', 'face2paint', 
    size=512, #device="cuda",
    side_by_side=False
)

def inference(img):
  return face2paint(model, img)

gradio.Interface(inference,
                 inputs=gradio.Image(type='pil'),
                 outputs=gradio.Image(type='pil'),
                 Instructuction=Instructuction, title=title, description=description, article=article,
                 examples=['Upload Xty.png',
                           'Upload Tg.png']).launch()