Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -106,9 +106,11 @@ 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: str = 1e9,
|
| 110 |
use_flash_decoder: bool = False,
|
| 111 |
rmbg: bool = True):
|
|
|
|
|
|
|
|
|
|
| 112 |
"""
|
| 113 |
Generate 3D part meshes from an input image.
|
| 114 |
"""
|
|
@@ -183,7 +185,6 @@ def build_demo():
|
|
| 183 |
num_tokens = gr.Slider(256, 2048, value=1024, step=64, label="Num Tokens")
|
| 184 |
num_steps = gr.Slider(1, 100, value=50, step=1, label="Inference Steps")
|
| 185 |
guidance = gr.Slider(1.0, 20.0, value=7.0, step=0.1, label="Guidance Scale")
|
| 186 |
-
max_coords = gr.Text(value="1e9", label="Max Expanded Coords")
|
| 187 |
flash_decoder = gr.Checkbox(value=False, label="Use Flash Decoder")
|
| 188 |
remove_bg = gr.Checkbox(value=False, label="Remove Background (RMBG)")
|
| 189 |
run_button = gr.Button("Generate 3D Parts")
|
|
@@ -199,7 +200,6 @@ def build_demo():
|
|
| 199 |
1024,
|
| 200 |
50,
|
| 201 |
7.0,
|
| 202 |
-
1e9,
|
| 203 |
False,
|
| 204 |
True
|
| 205 |
],
|
|
@@ -214,7 +214,7 @@ def build_demo():
|
|
| 214 |
|
| 215 |
run_button.click(fn=run_triposg,
|
| 216 |
inputs=[input_image, num_parts, seed, num_tokens, num_steps,
|
| 217 |
-
guidance,
|
| 218 |
outputs=[output_model, output_dir])
|
| 219 |
return demo
|
| 220 |
|
|
|
|
| 106 |
num_tokens: int = 1024,
|
| 107 |
num_inference_steps: int = 50,
|
| 108 |
guidance_scale: float = 7.0,
|
|
|
|
| 109 |
use_flash_decoder: bool = False,
|
| 110 |
rmbg: bool = True):
|
| 111 |
+
|
| 112 |
+
max_num_expanded_coords = 1e9
|
| 113 |
+
|
| 114 |
"""
|
| 115 |
Generate 3D part meshes from an input image.
|
| 116 |
"""
|
|
|
|
| 185 |
num_tokens = gr.Slider(256, 2048, value=1024, step=64, label="Num Tokens")
|
| 186 |
num_steps = gr.Slider(1, 100, value=50, step=1, label="Inference Steps")
|
| 187 |
guidance = gr.Slider(1.0, 20.0, value=7.0, step=0.1, label="Guidance Scale")
|
|
|
|
| 188 |
flash_decoder = gr.Checkbox(value=False, label="Use Flash Decoder")
|
| 189 |
remove_bg = gr.Checkbox(value=False, label="Remove Background (RMBG)")
|
| 190 |
run_button = gr.Button("Generate 3D Parts")
|
|
|
|
| 200 |
1024,
|
| 201 |
50,
|
| 202 |
7.0,
|
|
|
|
| 203 |
False,
|
| 204 |
True
|
| 205 |
],
|
|
|
|
| 214 |
|
| 215 |
run_button.click(fn=run_triposg,
|
| 216 |
inputs=[input_image, num_parts, seed, num_tokens, num_steps,
|
| 217 |
+
guidance, flash_decoder, remove_bg],
|
| 218 |
outputs=[output_model, output_dir])
|
| 219 |
return demo
|
| 220 |
|