Spaces:
Running
on
Zero
Running
on
Zero
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 |
|