app.py

import shlex
import subprocess


subprocess.run(shlex.split("pip install pip==24.0"), check=True)
subprocess.run(
    shlex.split(
        "pip install package/onnxruntime_gpu-1.17.0-cp310-cp310-manylinux_2_28_x86_64.whl --force-reinstall --no-deps"
    ), check=True
)
subprocess.run(
    shlex.split(
        "pip install package/nvdiffrast-0.3.1.torch-cp310-cp310-linux_x86_64.whl --force-reinstall --no-deps"
    ), check=True
)


if __name__ == "__main__":
    from huggingface_hub import snapshot_download

    snapshot_download("cavargas10/Unique3D", repo_type="model", local_dir="./ckpt")

    import os
    import sys
    sys.path.append(os.curdir)
    import torch
    torch.set_float32_matmul_precision('medium')
    torch.backends.cuda.matmul.allow_tf32 = True
    torch.set_grad_enabled(False)

import fire
import gradio as gr
from gradio_app.gradio_3dgen import create_ui as create_3d_ui
from gradio_app.all_models import model_zoo

_TITLE = """UTPL - Conversión de Imagen a objetos 3D usando IA"""
_DESCRIPTION = """
### Tesis: *"Objetos tridimensionales creados por IA: Innovación en entornos virtuales"*  
**Autor:** Carlos Vargas  
**Base técnica:** Adaptación de [Unique3D](https://wukailu.github.io/Unique3D/) (herramientas de código abierto para generación 3D)  
**Propósito educativo:** Demostraciones académicas e Investigación en modelado 3D automático    
**Agraecimiento especial a @hysts por su apoyo para el funcionamiento del demo**
"""

def launch():
    model_zoo.init_models()
        
    with gr.Blocks(
        title=_TITLE,
    ) as demo:
        with gr.Row():
            with gr.Column(scale=1):
                gr.Markdown('# ' + _TITLE)
        gr.Markdown(_DESCRIPTION)
        create_3d_ui("wkl") 
    demo.queue().launch(share=True)
    
if __name__ == '__main__':
    fire.Fire(launch)