Spaces:
Running
on
Zero
Running
on
Zero
xinjie.wang
commited on
Commit
·
81e9969
1
Parent(s):
38358b4
test
Browse files
app.py
CHANGED
@@ -1,6 +1,7 @@
|
|
1 |
import os
|
2 |
import shutil
|
3 |
from functools import partial
|
|
|
4 |
|
5 |
import gradio as gr
|
6 |
from common import (
|
@@ -17,7 +18,7 @@ from common import (
|
|
17 |
select_point,
|
18 |
)
|
19 |
from gradio.themes import Default
|
20 |
-
from gradio.themes.utils.colors import slate
|
21 |
from gradio_litmodel3d import LitModel3D
|
22 |
from asset3d_gen.models.delight import DelightingModel
|
23 |
from asset3d_gen.models.segment import RembgRemover, SAMPredictor
|
@@ -64,7 +65,7 @@ def end_session(req: gr.Request) -> None:
|
|
64 |
|
65 |
|
66 |
with gr.Blocks(
|
67 |
-
delete_cache=(43200, 43200), theme=Default(primary_hue=
|
68 |
) as demo:
|
69 |
gr.Markdown(
|
70 |
f"""
|
|
|
1 |
import os
|
2 |
import shutil
|
3 |
from functools import partial
|
4 |
+
import spaces
|
5 |
|
6 |
import gradio as gr
|
7 |
from common import (
|
|
|
18 |
select_point,
|
19 |
)
|
20 |
from gradio.themes import Default
|
21 |
+
from gradio.themes.utils.colors import slate, sky, blue, indigo
|
22 |
from gradio_litmodel3d import LitModel3D
|
23 |
from asset3d_gen.models.delight import DelightingModel
|
24 |
from asset3d_gen.models.segment import RembgRemover, SAMPredictor
|
|
|
65 |
|
66 |
|
67 |
with gr.Blocks(
|
68 |
+
delete_cache=(43200, 43200), theme=Default(primary_hue=indigo)
|
69 |
) as demo:
|
70 |
gr.Markdown(
|
71 |
f"""
|
common.py
CHANGED
@@ -4,7 +4,6 @@ import os
|
|
4 |
import sys
|
5 |
from glob import glob
|
6 |
from typing import Union
|
7 |
-
|
8 |
import cv2
|
9 |
import gradio as gr
|
10 |
import numpy as np
|
@@ -17,8 +16,6 @@ from kolors.pipelines.pipeline_stable_diffusion_xl_chatglm_256 import (
|
|
17 |
from kolors.pipelines.pipeline_stable_diffusion_xl_chatglm_256_ipadapter import ( # noqa
|
18 |
StableDiffusionXLPipeline as StableDiffusionXLPipelineIP,
|
19 |
)
|
20 |
-
import spaces
|
21 |
-
|
22 |
from PIL import Image
|
23 |
from tqdm import tqdm
|
24 |
from asset3d_gen.data.backproject_v2 import entrypoint as backproject_api
|
@@ -47,9 +44,7 @@ from asset3d_gen.validators.quality_checkers import (
|
|
47 |
from asset3d_gen.validators.urdf_convertor import URDFGenerator, zip_files
|
48 |
|
49 |
current_directory = os.getcwd()
|
50 |
-
|
51 |
-
sys.path.insert(0, current_directory)
|
52 |
-
|
53 |
from thirdparty.TRELLIS.trellis.pipelines import TrellisImageTo3DPipeline
|
54 |
from thirdparty.TRELLIS.trellis.renderers.mesh_renderer import MeshRenderer
|
55 |
from thirdparty.TRELLIS.trellis.representations import (
|
@@ -62,7 +57,6 @@ from thirdparty.TRELLIS.trellis.utils.render_utils import (
|
|
62 |
yaw_pitch_r_fov_to_extrinsics_intrinsics,
|
63 |
)
|
64 |
|
65 |
-
|
66 |
logging.basicConfig(
|
67 |
format="%(asctime)s - %(levelname)s - %(message)s", level=logging.INFO
|
68 |
)
|
@@ -70,6 +64,7 @@ logger = logging.getLogger(__name__)
|
|
70 |
|
71 |
|
72 |
MAX_SEED = 100000
|
|
|
73 |
|
74 |
|
75 |
@spaces.GPU
|
|
|
4 |
import sys
|
5 |
from glob import glob
|
6 |
from typing import Union
|
|
|
7 |
import cv2
|
8 |
import gradio as gr
|
9 |
import numpy as np
|
|
|
16 |
from kolors.pipelines.pipeline_stable_diffusion_xl_chatglm_256_ipadapter import ( # noqa
|
17 |
StableDiffusionXLPipeline as StableDiffusionXLPipelineIP,
|
18 |
)
|
|
|
|
|
19 |
from PIL import Image
|
20 |
from tqdm import tqdm
|
21 |
from asset3d_gen.data.backproject_v2 import entrypoint as backproject_api
|
|
|
44 |
from asset3d_gen.validators.urdf_convertor import URDFGenerator, zip_files
|
45 |
|
46 |
current_directory = os.getcwd()
|
47 |
+
sys.path.insert(0, current_directory)
|
|
|
|
|
48 |
from thirdparty.TRELLIS.trellis.pipelines import TrellisImageTo3DPipeline
|
49 |
from thirdparty.TRELLIS.trellis.renderers.mesh_renderer import MeshRenderer
|
50 |
from thirdparty.TRELLIS.trellis.representations import (
|
|
|
57 |
yaw_pitch_r_fov_to_extrinsics_intrinsics,
|
58 |
)
|
59 |
|
|
|
60 |
logging.basicConfig(
|
61 |
format="%(asctime)s - %(levelname)s - %(message)s", level=logging.INFO
|
62 |
)
|
|
|
64 |
|
65 |
|
66 |
MAX_SEED = 100000
|
67 |
+
os.environ["GRADIO_ANALYTICS_ENABLED"] = "false"
|
68 |
|
69 |
|
70 |
@spaces.GPU
|