sd-spritesheets / app.py
ronvolutional's picture
feat: Enable API
bab3cac
from typing import List
import time
import gradio as gr
from modules.sprites import generate_sides, build_gifs, build_spritesheet
def generate(prompt, thresh):
timestamp = int(time.time())
sides = generate_sides(prompt, 3)[0]
spritesheet = build_spritesheet(sides, prompt, timestamp=timestamp, thresh=thresh)[0]
filepaths = build_gifs(sides, prompt, save=True, timestamp=timestamp, thresh=thresh)[1]
return spritesheet, filepaths[0], filepaths[1], filepaths[2], filepaths[3]
demo = gr.Blocks()
with demo:
gr.Markdown("""
# Stable Diffusion Sprite Sheets
Generate a sprite sheet of pixel art character sides and their corresponding walk animations! Checkpoint by [Onodofthenorth](https://huggingface.co/Onodofthenorth/SD_PixelArt_SpriteSheet_Generator). Sprites are 32x32 pixels scaled up to 96x96. NSFW content replaced with blank sprites.
""")
with gr.Row():
with gr.Column():
prompt = gr.Textbox(label="Prompt", placeholder="Enter text prompt")
threshold = gr.Slider(label="Background removal amount", placeholder="Tweak how strong the background removal is", minimum=0, maximum=255, value=96)
button = gr.Button("Generate")
with gr.Box():
with gr.Row():
spritesheet = gr.Image(label="Sprite Sheet")
with gr.Row():
front = gr.Image(label="Front")
back = gr.Image(label="Back")
left = gr.Image(label="Left")
right = gr.Image(label="Right")
button.click(fn=generate, inputs=[prompt, threshold], outputs=[spritesheet, front, back, left, right])
demo.queue()
demo.launch(show_api=True)