Spaces:
Running
on
T4
Running
on
T4
Upload gradio_app.py
Browse files- gradio_app.py +20 -82
gradio_app.py
CHANGED
@@ -362,29 +362,13 @@ def shape_generation(
|
|
362 |
|
363 |
|
364 |
def build_app():
|
365 |
-
title = '
|
366 |
-
if MV_MODE:
|
367 |
-
title = 'Hunyuan3D-2mv: Image to 3D Generation with 1-4 Views'
|
368 |
-
if 'mini' in args.subfolder:
|
369 |
-
title = 'Hunyuan3D-2mini: Strong 0.6B Image to Shape Generator'
|
370 |
-
if TURBO_MODE:
|
371 |
-
title = title.replace(':', '-Turbo: Fast ')
|
372 |
|
373 |
title_html = f"""
|
374 |
<div style="font-size: 2em; font-weight: bold; text-align: center; margin-bottom: 5px">
|
375 |
|
376 |
{title}
|
377 |
</div>
|
378 |
-
<div align="center">
|
379 |
-
Tencent Hunyuan3D Team
|
380 |
-
</div>
|
381 |
-
<div align="center">
|
382 |
-
<a href="https://github.com/tencent/Hunyuan3D-2">Github</a>  
|
383 |
-
<a href="http://3d-models.hunyuan.tencent.com">Homepage</a>  
|
384 |
-
<a href="https://3d.hunyuan.tencent.com">Hunyuan3D Studio</a>  
|
385 |
-
<a href="#">Technical Report</a>  
|
386 |
-
<a href="https://huggingface.co/Tencent/Hunyuan3D-2"> Pretrained Models</a>  
|
387 |
-
</div>
|
388 |
"""
|
389 |
custom_css = """
|
390 |
.app.svelte-wpkpf6.svelte-wpkpf6:not(.fill_width) {
|
@@ -405,26 +389,17 @@ def build_app():
|
|
405 |
|
406 |
with gr.Row():
|
407 |
with gr.Column(scale=3):
|
408 |
-
with gr.
|
409 |
-
with gr.
|
410 |
-
|
411 |
-
|
412 |
-
|
413 |
-
|
414 |
-
|
415 |
-
|
416 |
-
|
417 |
-
|
418 |
-
|
419 |
-
mv_image_front = gr.Image(label='Front', type='pil', image_mode='RGBA', height=140,
|
420 |
-
min_width=100, elem_classes='mv-image')
|
421 |
-
mv_image_back = gr.Image(label='Back', type='pil', image_mode='RGBA', height=140,
|
422 |
-
min_width=100, elem_classes='mv-image')
|
423 |
-
with gr.Row():
|
424 |
-
mv_image_left = gr.Image(label='Left', type='pil', image_mode='RGBA', height=140,
|
425 |
-
min_width=100, elem_classes='mv-image')
|
426 |
-
mv_image_right = gr.Image(label='Right', type='pil', image_mode='RGBA', height=140,
|
427 |
-
min_width=100, elem_classes='mv-image')
|
428 |
|
429 |
with gr.Row():
|
430 |
btn = gr.Button(value='Gen Shape', variant='primary', min_width=100)
|
@@ -482,7 +457,7 @@ def build_app():
|
|
482 |
file_export = gr.DownloadButton(label="Download", variant='primary',
|
483 |
interactive=False, min_width=100)
|
484 |
|
485 |
-
with gr.Column(scale=
|
486 |
with gr.Tabs(selected='gen_mesh_panel') as tabs_output:
|
487 |
with gr.Tab('Generated Mesh', id='gen_mesh_panel'):
|
488 |
html_gen_mesh = gr.HTML(HTML_OUTPUT_PLACEHOLDER, label='Output')
|
@@ -491,48 +466,6 @@ def build_app():
|
|
491 |
with gr.Tab('Mesh Statistic', id='stats_panel'):
|
492 |
stats = gr.Json({}, label='Mesh Stats')
|
493 |
|
494 |
-
with gr.Column(scale=3 if MV_MODE else 2):
|
495 |
-
with gr.Tabs(selected='tab_img_gallery') as gallery:
|
496 |
-
with gr.Tab('Image to 3D Gallery', id='tab_img_gallery', visible=not MV_MODE) as tab_gi:
|
497 |
-
with gr.Row():
|
498 |
-
gr.Examples(examples=example_is, inputs=[image],
|
499 |
-
label=None, examples_per_page=18)
|
500 |
-
|
501 |
-
with gr.Tab('Text to 3D Gallery', id='tab_txt_gallery', visible=HAS_T2I and not MV_MODE) as tab_gt:
|
502 |
-
with gr.Row():
|
503 |
-
gr.Examples(examples=example_ts, inputs=[caption],
|
504 |
-
label=None, examples_per_page=18)
|
505 |
-
with gr.Tab('MultiView to 3D Gallery', id='tab_mv_gallery', visible=MV_MODE) as tab_mv:
|
506 |
-
with gr.Row():
|
507 |
-
gr.Examples(examples=example_mvs,
|
508 |
-
inputs=[mv_image_front, mv_image_back, mv_image_left, mv_image_right],
|
509 |
-
label=None, examples_per_page=6)
|
510 |
-
|
511 |
-
gr.HTML(f"""
|
512 |
-
<div align="center">
|
513 |
-
Activated Model - Shape Generation ({args.model_path}/{args.subfolder}) ; Texture Generation ({'Hunyuan3D-2' if HAS_TEXTUREGEN else 'Unavailable'})
|
514 |
-
</div>
|
515 |
-
""")
|
516 |
-
if not HAS_TEXTUREGEN:
|
517 |
-
gr.HTML("""
|
518 |
-
<div style="margin-top: 5px;" align="center">
|
519 |
-
<b>Warning: </b>
|
520 |
-
Texture synthesis is disable due to missing requirements,
|
521 |
-
please install requirements following <a href="https://github.com/Tencent/Hunyuan3D-2?tab=readme-ov-file#install-requirements">README.md</a>to activate it.
|
522 |
-
</div>
|
523 |
-
""")
|
524 |
-
if not args.enable_t23d:
|
525 |
-
gr.HTML("""
|
526 |
-
<div style="margin-top: 5px;" align="center">
|
527 |
-
<b>Warning: </b>
|
528 |
-
Text to 3D is disable. To activate it, please run `python gradio_app.py --enable_t23d`.
|
529 |
-
</div>
|
530 |
-
""")
|
531 |
-
|
532 |
-
tab_ip.select(fn=lambda: gr.update(selected='tab_img_gallery'), outputs=gallery)
|
533 |
-
if HAS_T2I:
|
534 |
-
tab_tp.select(fn=lambda: gr.update(selected='tab_txt_gallery'), outputs=gallery)
|
535 |
-
|
536 |
btn.click(
|
537 |
shape_generation,
|
538 |
inputs=[
|
@@ -671,6 +604,12 @@ if __name__ == '__main__':
|
|
671 |
parser.add_argument('--low_vram_mode', action='store_true')
|
672 |
args = parser.parse_args()
|
673 |
|
|
|
|
|
|
|
|
|
|
|
|
|
674 |
args.enable_flashvdm = True
|
675 |
SAVE_DIR = args.cache_path
|
676 |
os.makedirs(SAVE_DIR, exist_ok=True)
|
@@ -684,8 +623,7 @@ if __name__ == '__main__':
|
|
684 |
HTML_OUTPUT_PLACEHOLDER = f"""
|
685 |
<div style='height: {650}px; width: 100%; border-radius: 8px; border-color: #e5e7eb; border-style: solid; border-width: 1px; display: flex; justify-content: center; align-items: center;'>
|
686 |
<div style='text-align: center; font-size: 16px; color: #6b7280;'>
|
687 |
-
<p style="color: #8d8d8d;">
|
688 |
-
<p style="color: #8d8d8d;">No mesh here.</p>
|
689 |
</div>
|
690 |
</div>
|
691 |
"""
|
|
|
362 |
|
363 |
|
364 |
def build_app():
|
365 |
+
title = 'Generación de modelo basado en imágenes (de 1 a 4 vistas)'
|
|
|
|
|
|
|
|
|
|
|
|
|
366 |
|
367 |
title_html = f"""
|
368 |
<div style="font-size: 2em; font-weight: bold; text-align: center; margin-bottom: 5px">
|
369 |
|
370 |
{title}
|
371 |
</div>
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
372 |
"""
|
373 |
custom_css = """
|
374 |
.app.svelte-wpkpf6.svelte-wpkpf6:not(.fill_width) {
|
|
|
389 |
|
390 |
with gr.Row():
|
391 |
with gr.Column(scale=3):
|
392 |
+
with gr.Row():
|
393 |
+
with gr.Row():
|
394 |
+
mv_image_front = gr.Image(label='Front', type='pil', image_mode='RGBA', height=140,
|
395 |
+
min_width=100, elem_classes='mv-image')
|
396 |
+
mv_image_back = gr.Image(label='Back', type='pil', image_mode='RGBA', height=140,
|
397 |
+
min_width=100, elem_classes='mv-image')
|
398 |
+
with gr.Row():
|
399 |
+
mv_image_left = gr.Image(label='Left', type='pil', image_mode='RGBA', height=140,
|
400 |
+
min_width=100, elem_classes='mv-image')
|
401 |
+
mv_image_right = gr.Image(label='Right', type='pil', image_mode='RGBA', height=140,
|
402 |
+
min_width=100, elem_classes='mv-image')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
403 |
|
404 |
with gr.Row():
|
405 |
btn = gr.Button(value='Gen Shape', variant='primary', min_width=100)
|
|
|
457 |
file_export = gr.DownloadButton(label="Download", variant='primary',
|
458 |
interactive=False, min_width=100)
|
459 |
|
460 |
+
with gr.Column(scale=9):
|
461 |
with gr.Tabs(selected='gen_mesh_panel') as tabs_output:
|
462 |
with gr.Tab('Generated Mesh', id='gen_mesh_panel'):
|
463 |
html_gen_mesh = gr.HTML(HTML_OUTPUT_PLACEHOLDER, label='Output')
|
|
|
466 |
with gr.Tab('Mesh Statistic', id='stats_panel'):
|
467 |
stats = gr.Json({}, label='Mesh Stats')
|
468 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
469 |
btn.click(
|
470 |
shape_generation,
|
471 |
inputs=[
|
|
|
604 |
parser.add_argument('--low_vram_mode', action='store_true')
|
605 |
args = parser.parse_args()
|
606 |
|
607 |
+
try:
|
608 |
+
port = int(args.port)
|
609 |
+
except ValueError:
|
610 |
+
print(f"Invalid port argument detected: {args.port} — using default 7860")
|
611 |
+
port = 7860
|
612 |
+
|
613 |
args.enable_flashvdm = True
|
614 |
SAVE_DIR = args.cache_path
|
615 |
os.makedirs(SAVE_DIR, exist_ok=True)
|
|
|
623 |
HTML_OUTPUT_PLACEHOLDER = f"""
|
624 |
<div style='height: {650}px; width: 100%; border-radius: 8px; border-color: #e5e7eb; border-style: solid; border-width: 1px; display: flex; justify-content: center; align-items: center;'>
|
625 |
<div style='text-align: center; font-size: 16px; color: #6b7280;'>
|
626 |
+
<p style="color: #8d8d8d;">Bienvenido a 3D Market</p>
|
|
|
627 |
</div>
|
628 |
</div>
|
629 |
"""
|