Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -106,21 +106,13 @@ def run_triposg(image_path: str,
|
|
106 |
num_tokens: int = 1024,
|
107 |
num_inference_steps: int = 50,
|
108 |
guidance_scale: float = 7.0,
|
109 |
-
max_num_expanded_coords:
|
110 |
use_flash_decoder: bool = False,
|
111 |
rmbg: bool = True):
|
112 |
"""
|
113 |
Generate 3D part meshes from an input image.
|
114 |
"""
|
115 |
|
116 |
-
num_parts = 3
|
117 |
-
seed = 0
|
118 |
-
num_tokens = 1024
|
119 |
-
num_inference_steps = 50
|
120 |
-
guidance_scale = 7
|
121 |
-
max_num_expanded_coords = 1e9
|
122 |
-
use_flash_decoder = False
|
123 |
-
|
124 |
if rmbg:
|
125 |
img_pil = prepare_image(image_path, bg_color=np.array([1.0, 1.0, 1.0]), rmbg_net=rmbg_net)
|
126 |
else:
|
|
|
106 |
num_tokens: int = 1024,
|
107 |
num_inference_steps: int = 50,
|
108 |
guidance_scale: float = 7.0,
|
109 |
+
max_num_expanded_coords: str = 1e9,
|
110 |
use_flash_decoder: bool = False,
|
111 |
rmbg: bool = True):
|
112 |
"""
|
113 |
Generate 3D part meshes from an input image.
|
114 |
"""
|
115 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
116 |
if rmbg:
|
117 |
img_pil = prepare_image(image_path, bg_color=np.array([1.0, 1.0, 1.0]), rmbg_net=rmbg_net)
|
118 |
else:
|