Spaces:
Build error
Build error
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:
|