Spaces:
Running
on
Zero
Running
on
Zero
File size: 5,562 Bytes
224d97a 51fbf43 224d97a 8fd7005 224d97a 9e68ed7 224d97a 9c24db9 |
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 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 |
import gradio as gr
import spaces
from omni_zero import OmniZeroSingle
@spaces.GPU(duration=180)
def generate(
seed=42,
prompt="A person",
negative_prompt="blurry, out of focus",
guidance_scale=3.0,
number_of_images=1,
number_of_steps=10,
base_image="https://github.com/okaris/omni-zero/assets/1448702/2ca63443-c7f3-4ba6-95c1-2a341414865f",
base_image_strength=0.15,
composition_image="https://github.com/okaris/omni-zero/assets/1448702/2ca63443-c7f3-4ba6-95c1-2a341414865f",
composition_image_strength=1.0,
style_image="https://github.com/okaris/omni-zero/assets/1448702/64dc150b-f683-41b1-be23-b6a52c771584",
style_image_strength=1.0,
identity_image="https://github.com/okaris/omni-zero/assets/1448702/ba193a3a-f90e-4461-848a-560454531c58",
identity_image_strength=1.0,
depth_image=None,
depth_image_strength=0.5,
):
omni_zero = OmniZeroSingle(
base_model="frankjoshua/albedobaseXL_v13",
)
images = omni_zero.generate(
seed=seed,
prompt=prompt,
negative_prompt=negative_prompt,
guidance_scale=guidance_scale,
number_of_images=number_of_images,
number_of_steps=number_of_steps,
base_image=base_image,
base_image_strength=base_image_strength,
composition_image=composition_image,
composition_image_strength=composition_image_strength,
style_image=style_image,
style_image_strength=style_image_strength,
identity_image=identity_image,
identity_image_strength=identity_image_strength,
depth_image=depth_image,
depth_image_strength=depth_image_strength,
)
# for i, image in enumerate(images):
# image.save(f"oz_output_{i}.jpg")
return images
with gr.Blocks() as demo:
with gr.Row():
with gr.Column():
with gr.Row():
prompt = gr.Textbox(label="Prompt", value="A person")
with gr.Row():
negative_prompt = gr.Textbox(label="Negative Prompt", value="blurry, out of focus")
with gr.Row():
seed = gr.Slider(label="Seed",step=1, minimum=0, maximum=10000000, value=42)
number_of_images = gr.Slider(label="Number of Outputs",step=1, minimum=1, maximum=4, value=1)
with gr.Row():
guidance_scale = gr.Slider(label="Guidance Scale",step=0.1, minimum=0.0, maximum=14.0, value=3.0)
number_of_steps = gr.Slider(label="Number of Steps",step=1, minimum=1, maximum=50, value=10)
with gr.Row():
with gr.Column():
with gr.Row():
base_image = gr.Image(label="Base Image", value="https://github.com/okaris/omni-zero/assets/1448702/2ca63443-c7f3-4ba6-95c1-2a341414865f")
with gr.Row():
base_image_strength = gr.Slider(label="Base Image Strength",step=0.01, minimum=0.0, maximum=1.0, value=0.15)
with gr.Column():
with gr.Row():
composition_image = gr.Image(label="Composition", value="https://github.com/okaris/omni-zero/assets/1448702/2ca63443-c7f3-4ba6-95c1-2a341414865f")
with gr.Row():
composition_image_strength = gr.Slider(label="Composition Image Strength",step=0.01, minimum=0.0, maximum=1.0, value=1.0)
# with gr.Row():
with gr.Column():
with gr.Row():
style_image = gr.Image(label="Style Image", value="https://github.com/okaris/omni-zero/assets/1448702/64dc150b-f683-41b1-be23-b6a52c771584")
with gr.Row():
style_image_strength = gr.Slider(label="Style Image Strength",step=0.01, minimum=0.0, maximum=1.0, value=1.0)
with gr.Column():
with gr.Row():
identity_image = gr.Image(label="Identity Image", value="https://github.com/okaris/omni-zero/assets/1448702/ba193a3a-f90e-4461-848a-560454531c58")
with gr.Row():
identity_image_strength = gr.Slider(label="Identitiy Image Strenght",step=0.01, minimum=0.0, maximum=1.0, value=1.0)
# with gr.Column():
# with gr.Row():
# depth_image = gr.Image(label="depth_image", value=None)
# with gr.Row():
# depth_image_strength = gr.Slider(label="depth_image_strength",step=0.01, minimum=0.0, maximum=1.0, value=0.5)
with gr.Column():
with gr.Row():
out = gr.Gallery(label="Output(s)")
with gr.Row():
# clear = gr.Button("Clear")
submit = gr.Button("Generate")
submit.click(generate, inputs=[
seed,
prompt,
negative_prompt,
guidance_scale,
number_of_images,
number_of_steps,
base_image,
base_image_strength,
composition_image,
composition_image_strength,
style_image,
style_image_strength,
identity_image,
identity_image_strength,
],
outputs=[out]
)
# clear.click(lambda: None, None, chatbot, queue=False)
if __name__ == "__main__":
demo.launch() |