xinjie.wang commited on
Commit
81e9969
·
1 Parent(s): 38358b4
Files changed (2) hide show
  1. app.py +3 -2
  2. common.py +2 -7
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=slate)
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
- if current_directory not in sys.path:
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