Spaces:
Runtime error
Runtime error
File size: 4,771 Bytes
eca813c 30e735a eca813c 30e735a eca813c 30e735a eca813c 2e4fc42 eca813c 2e4fc42 30e735a eca813c 30e735a eca813c 30e735a eca813c 726ebdf eca813c 30e735a eca813c 95b7555 30e735a 726ebdf 95b7555 eca813c |
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 |
import gradio as gr
import os
from model import VirtualStagingToolV2
def predict(image, style, backyard_style, color_preference):
init_image = image.convert("RGB").resize((512, 512))
vs_tool = VirtualStagingToolV2(diffusion_version="stabilityai/stable-diffusion-2-inpainting")
if backyard_style:
style = backyard_style
output_images, transparent_mask_image = vs_tool.virtual_stage(
image=init_image, style=style, color_preference=color_preference, number_images=1)
return output_images[0], transparent_mask_image
image_blocks = gr.Blocks()
with image_blocks as demo:
gr.Markdown(
"""
# Virtual Home Staging
""")
with gr.Group():
with gr.Box():
with gr.Row():
with gr.Column():
image = gr.Image(source='upload', elem_id="image_upload",
type="pil", label="Upload",
).style(height=400)
with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
style = gr.Dropdown(
["Bohemian", "Coastal", "Contemporary", "Farmhouse", "French country",
"Glam", "Industrial", "Japandi", "Mid-century modern", "Minimal", "Modern",
"Rustic", "Scandinavian", "Traditional", "Transitional", ],
label="Design theme", elem_id="input-style",
info="only select if the image is NOT a backyard"
)
backyard_style = gr.Dropdown(
["Beautiful garden", "Charming playhouse garden landscaping",
"Cottage garden's colorful planting palette", "Cozy corner with fire pit and seating",
"Garden landscaping with gravel landscaping", "Hip california garden landscaping",
"Lush green lawn", "Mediterranean garden landscaping",
"Moss garden", "Outdoor dining and sitting area",
"Party-ready outdoor space with pool, spa, and fire feature",
"Resort-style landscaping and pool", "Round swimming pool with lawn and pool house"],
label="Backyard design theme", elem_id="input-backyard-style",
info="only select if the image is a backyard"
)
color_preference = gr.Textbox(placeholder='Enter color preference',
label="Color preference", elem_id="input-color")
btn = gr.Button("Inpaint!").style(
margin=False,
rounded=(False, True, True, False),
full_width=False,
)
with gr.Column():
mask_image = gr.Image(label="Mask image", elem_id="mask-img", type="pil").style(height=800)
image_out = gr.Image(label="Output", elem_id="output-img", type="pil").style(height=800)
btn.click(fn=predict, inputs=[image, style, backyard_style, color_preference], outputs=[image_out, mask_image])
gr.Markdown("## Image Examples")
gr.Examples(
examples=[os.path.join(os.path.dirname(__file__),
"examples/exciting-small-kitchen-ideas-1821197-hero-d00f516e2fbb4dcabb076ee9685e877a.jpg"),
os.path.join(os.path.dirname(__file__),
"examples/modern-kitchen-cabinets.jpg"),
os.path.join(os.path.dirname(__file__),
"examples/oct-2019-idh-bathroom-reno-ideas-new-gallery-2.jpg"),
os.path.join(os.path.dirname(__file__),
"examples/tips-for-decorating-a-beautiful-bedroom-1976169-hero-e960fbb8311c4b9b875a1813962d34eb.jpg"),
os.path.join(os.path.dirname(__file__),
"examples/backyard_additions.jpg"),
os.path.join(os.path.dirname(__file__),
"examples/living-room-gallery-shelves-l-shaped-couch-ELeyNpyyqpZ8hosOG3EG1X-b5a39646574544e8a75f2961332cd89a.jpg"),
os.path.join(os.path.dirname(__file__),
"examples/modern-dining-room-ideas-4147451-hero-d6333998f8b34620adfd4d99ac732586.jpg"),
],
inputs=image
)
image_blocks.launch() |