pixel-art-style / app.py
kohbanye's picture
add description
e40c467
raw
history blame contribute delete
489 Bytes
import gradio as gr
import diffusers
from PIL import Image
pipe = diffusers.DiffusionPipeline.from_pretrained("kohbanye/pixel-art-style")
def generate_image(prompt: str) -> Image.Image:
image = pipe(prompt, num_inference_steps=50, width=512, height=512).images[0]
return image
app = gr.Interface(
fn=generate_image,
inputs="text",
outputs="image",
title="Pixel Art Generator",
description="Use 'pixelartstyle' token to generate pixel art.",
)
app.launch()