Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -112,6 +112,15 @@ def run_triposg(image_path: str,
|
|
| 112 |
"""
|
| 113 |
Generate 3D part meshes from an input image.
|
| 114 |
"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 115 |
if rmbg:
|
| 116 |
img_pil = prepare_image(image_path, bg_color=np.array([1.0, 1.0, 1.0]), rmbg_net=rmbg_net)
|
| 117 |
else:
|
|
|
|
| 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:
|