ashawkey commited on
Commit
1a5d02b
1 Parent(s): a93901d
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -11,12 +11,16 @@ from safetensors.torch import load_file
11
  import rembg
12
  import gradio as gr
13
 
 
 
 
 
 
 
14
  import kiui
15
  from kiui.op import recenter
16
  from kiui.cam import orbit_camera
17
 
18
- os.system('pip install -e ./diff-gaussian-rasterization')
19
-
20
  from core.options import AllConfigs, Options
21
  from core.models import LGM
22
  from mvdream.pipeline_mvdream import MVDreamPipeline
 
11
  import rembg
12
  import gradio as gr
13
 
14
+ try:
15
+ from diff_gaussian_rasterization import GaussianRasterizer
16
+ except ImportError:
17
+ os.system("pip install -e ./diff-gaussian-rasterization")
18
+ from diff_gaussian_rasterization import GaussianRasterizer
19
+
20
  import kiui
21
  from kiui.op import recenter
22
  from kiui.cam import orbit_camera
23
 
 
 
24
  from core.options import AllConfigs, Options
25
  from core.models import LGM
26
  from mvdream.pipeline_mvdream import MVDreamPipeline