Spaces:
Runtime error
Runtime error
File size: 5,101 Bytes
87f54b3 |
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 |
import gradio as gr
from PIL import Image
elements = {
"Becane": ("data/becane_1_0_2_014/before/img6", "data/becane_1_0_2_014/after/img7", "models/RoadBecane_f.stl"),
"BTWIN": ("data/btwin_1_26_3_114/before/img0", "data/btwin_1_26_3_114/after/img13", "models/BTWIN.stl"),
"Classic Road": ("data/croad_4_26_4_112/before/img3", "data/croad_4_26_4_112/after/img7", "models/ClassicRoad.stl"),
"Domane": ("data/domane_2_0_39_012/before/img6", "data/domane_2_0_39_012/after/img13", "models/domane.stl"),
"Enduro": ("data/enduro_2_18_18_105/before/img6", "data/enduro_2_18_18_105/after/img10", "models/enduro.stl"),
"G1": ("data/g1_2_36_14_010/before/img6", "data/g1_2_36_14_010/after/img8", "models/g1.stl"),
"GBike": ("data/gbike_3_31_8_018/before/img2", "data/gbike_3_31_8_018/after/img8", "models/gbike.stl"),
"Holland": (
"data/holland_4_24_18_00/before/img4", "data/holland_4_24_18_00/after/img7", "models/holland.stl"),
"Huffy": (
"data/huffy_6_4_35_012/before/img4", "data/huffy_6_4_35_012/after/img12", "models/huffy.stl"),
"Kuota": (
"data/kuota_6_5_30_112/before/img4", "data/kuota_6_5_30_112/after/img12", "models/kuota.stl"),
"MFactory": (
"data/mfactory_1_17_8_00/before/img4", "data/mfactory_1_17_8_00/after/img12", "models/mfactory.stl"),
"Mirage": (
"data/mirage_6_11_36_117/before/img4", "data/mirage_6_11_36_117/after/img12", "models/mirage.stl"),
"Old Bike": (
"data/oldbike_3_30_15_100/before/img4", "data/oldbike_3_30_15_100/after/img12", "models/oldbike.stl"),
"Freeride": (
"data/freeride_6_24_0_00/before/img4", "data/freeride_6_24_0_00/after/img12", "models/freeride_1Step_f.stl"),
"RondoRuut": (
"data/rondo_2_11_39_106/before/img4", "data/rondo_2_11_39_106/after/img12",
"models/RondoRuutClean_1Step_f.stl"),
"MTB Ghost": (
"data/ghost_1_0_4_116/before/img4", "data/ghost_1_0_4_116/after/img12", "models/MTB_Ghost.stl"),
"Scalpel": (
"data/scalpel_6_2_12_101/before/img4", "data/scalpel_6_2_12_101/after/img12", "models/scalpel.stl"),
"Verdona": (
"data/verdona_1_14_10_118/before/img4", "data/verdona_1_14_10_118/after/img12", "models/verdona.stl"),
"Vintage": (
"data/vintage_6_6_33_011/before/img4", "data/vintage_6_6_33_011/after/img12", "models/Vintage.stl"),
"WBike": (
"data/wbike_3_19_23_00/before/img4", "data/wbike_3_19_23_00/after/img12", "models/wbike.stl"),
}
def generate_ui(key):
img_b, img_a, model = elements[key]
gr.Markdown("### Before damage")
with gr.Row():
with gr.Row():
with gr.Column(scale=1):
gr.Markdown("#### Render")
img_bn = gr.Image(f"{img_b}.png")
with gr.Column(scale=1):
gr.Markdown("#### Background")
img_bb = gr.Image(f"{img_b}_background.png")
with gr.Column(scale=1):
gr.Markdown("#### Foreground")
img_bf = gr.Image(f"{img_b}_foreground.png")
with gr.Column(scale=1):
gr.Markdown("#### Segmentation")
img_bs = gr.Image(f"{img_b}_segmentation.png")
gr.Markdown("### After damage")
with gr.Row():
with gr.Column(scale=1):
gr.Markdown("#### Render")
img_an = gr.Image(f"{img_a}.png")
with gr.Column(scale=1):
gr.Markdown("#### Background")
img_ab = gr.Image(f"{img_a}_background.png")
with gr.Column(scale=1):
gr.Markdown("#### Foreground")
img_af = gr.Image(f"{img_a}_foreground.png")
with gr.Column(scale=1):
gr.Markdown("#### Segmentation")
img_as = gr.Image(f"{img_a}_segmentation.png")
model = gr.Model3D(model, label="3D model preview")
return img_bn, img_bb, img_bf, img_bs, img_an, img_ab, img_af, img_as, model
def get_values(key):
img_b, img_a, model = elements[key]
gr.Markdown("### Before damage")
img_bn = Image.open(f"{img_b}.png")
img_bb = Image.open(f"{img_b}_background.png")
img_bf = Image.open(f"{img_b}_foreground.png")
img_bs = Image.open(f"{img_b}_segmentation.png")
img_an = Image.open(f"{img_a}.png")
img_ab = Image.open(f"{img_a}_background.png")
img_af = Image.open(f"{img_a}_foreground.png")
img_as = Image.open(f"{img_a}_segmentation.png")
return img_bn, img_bb, img_bf, img_bs, img_an, img_ab, img_af, img_as, model
block = gr.Blocks()
with block:
with open("page.md", "r") as f:
gr.Markdown(f.read())
first_key = list(elements.keys())[0]
dropdown = gr.Dropdown(choices=list(elements.keys()), value=first_key, label="Model", interactive=True)
img_bn, img_bb, img_bf, img_bs, img_an, img_ab, img_af, img_as, model = generate_ui(first_key)
dropdown.change(get_values, inputs=[dropdown],
outputs=[img_bn, img_bb, img_bf, img_bs, img_an, img_ab, img_af,
img_as, model])
block.queue(concurrency_count=40, max_size=20).launch(max_threads=150)
|