Spaces:
Runtime error
Runtime error
File size: 1,556 Bytes
81ea62c fe1c2c9 81ea62c bab3cac |
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 40 41 42 43 44 45 46 47 48 49 |
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)
|